Pytorch中 bidirectional=True,Tensor的大小变化

1
2
3
4
5
6
7
8
9
10
11
12
13
rnn = nn.RNN(input_size=4,hidden_size=3,num_layers=2,batch_first=True, bidirectional = True)
input = torch.randn(1,5,4)
output , h_n = rnn(input)
print(output.shape)
print(h_n.shape)
 
 
#----输出结果----------
input.Size([1, 5, 4])
output.Size([1, 5, 6]) # hidden_size 要乘以2
h_n.Size([4, 1, 3]) # num_layers 要乘以2
 
seq_len = 1  num_layers = 2 hidden_size = 3 input_size = 4 btach_size = 5

  

posted @   华小电  阅读(313)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
点击右上角即可分享
微信分享提示