摘要:
Pytorch之permute函数 阅读全文
posted @ 2020-08-13 19:48
Picassooo
阅读(470)
评论(0)
推荐(0)
摘要:
torch.nn.lstm()接受的数据输入是(序列长度,batchsize,输入维数),使用batch_first=True,可以使lstm接受维度为(batchsize,序列长度,输入维数)的数据输入,同时,lstm的输出数据维度也会变为batchsize放在第一维(可参考这篇博客)。 阅读全文
posted @ 2020-08-13 12:03
Picassooo
阅读(2652)
评论(0)
推荐(0)

浙公网安备 33010602011771号