What is the shape of cudnn RNN input data?

I did not find a detailed description of the shape of cudnn RNN input data in the API documentation. I want to make sure that the input data shape of RNN is [batchSize, featureNum, timestep]. Am I right?

Hi,

Please refer to below sample:


Thanks

Thank you for your reply. The input data shape is [timeStep, batchSize, featureNum]. Is that right?

Yes, for RNN the tensor dims are [timeStep, batchSize, featureNum]

Thanks