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]