芭芭蘑菇

关注

torch.RNN使用

芭芭蘑菇

关注

阅读 68

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)

在这里插入图片描述

相关推荐

飞进科技

RNN使用示例

飞进科技 92 0 0

爱我中华8898

torch张量的使用

爱我中华8898 94 0 0

weipeng2k

使用RNN实现回归(16)

weipeng2k 58 0 0

yellowone

pytorch中torch.nn.utils.rnn相关sequence的pad和pack操作

yellowone 30 0 0

小禹说财

torch.reshape()的使用

小禹说财 83 0 0

小亦同学321

torch.max使用示例

小亦同学321 77 0 0

菜头粿子园

torch.where()的使用

菜头粿子园 68 0 0

止止_8fc8

torch的使用笔记

止止_8fc8 67 0 0

我阿霆哥

torch.nn中LSTM使用

我阿霆哥 132 0 0

Gaaidou

深度学习--RNN以及RNN的延伸

Gaaidou 32 0 0

精彩评论(0)

0 0 举报