1. 首页
  2. 数据库
  3. 其它
  4. 【PyTorch学习笔记】21:nn.RNN和nn.RNNCell的使用

【PyTorch学习笔记】21:nn.RNN和nn.RNNCell的使用

上传者: 2021-01-15 06:10:51上传 PDF文件 415.58KB 热度 11次
这节学习PyTorch的循环神经网络层nn.RNN,以及循环神经网络单元nn.RNNCell的一些细节。 1 nn.RNN涉及的Tensor PyTorch中的nn.RNN的数据处理如下图所示。每次向网络中输入batch个样本,每个时刻处理的是该时刻的batch个样本,因此xtx_txt​是shape为[batch,feature_len][batch, feature\_len][batch,feature_len]的Tensor。例如,输入3句话,每句话10个单词,每个单词用100维的向量表示,那么seq_len=10seq\_len=10seq_len=10,batch=3batch=3
下载地址
用户评论