torch.RNN使用

阅读 69

2022-02-05

inputs = torch.randn(3, 4, 16)
rnn = torch.nn.RNN(input_size=16, hidden_size=6, num_layers=20)
h0 = torch.randn(20, 4, 6)
outputs, _ = rnn(inputs, h0)
print(outputs)

在这里插入图片描述

相关推荐

精彩评论(0)

0 0 举报