Pytorch
文章平均质量分 62
demoluck
这个作者很懒,什么都没留下…
展开
-
Pytorch中LSTM网络参数
torch.nn.LSTM()输入API 重要参数 input_size: 每一个时步(time_step)输入到lstm单元的维度.(实际输入的数据size为[batch_size, input_size]) hidden_size: 确定了隐状态hidden_state的维度. 可以简单的看成: 构造了一个权重 , 隐含状态 num_layers: 叠加的层数。如图所示num_layers为3 batch_first: 输入数据的size为[batch_size, time_step,..转载 2021-10-08 23:05:55 · 670 阅读 · 0 评论 -
2021-10-08 pytorch中的常见函数
https://blog.csdn.net/xiongchengluo1129/article/details/79017142?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163133369716780274128034%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=1631333697167802741280原创 2021-10-08 22:24:41 · 121 阅读 · 0 评论