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?


Please refer to below sample:


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]