输入数据的维度
最近在复习RNN,记录一下
一般CNN的输入形状为:
<[batch_size(一次性运算的个数), channels(通道数), height(行数,或者说图片的高), width(列数,或者说图片的宽)]>
一般RNN的默认输入形状为:
<[seq_len(序列的长度,或者说时间的步数), batch_size(一次性运算的个数), input_dim(序列特征的维度)]>
所以在定义的时候建议设置batch_first=True。这样就省了再去换轴了
之后想起来了再补充更新,感谢阅读。