pytorch 中的 torch.nn.RNN 的参数

1、定义RNN的网络结构的参数(类似于CNN中定义 in_channel,out_channel,kernel_size等等)          input_size   输入x的特征大小(以mnist图像为例,特征大小为28*28 = 784)          hidden_size   隐藏层h的特征大小          num_layers    循环层的数量(RNN中重复的部分)  
相关文章
相关标签/搜索