- 博客(3)
- 收藏
- 关注
转载 Pytorch中LSTM网络参数
torch.nn.LSTM()输入API重要参数input_size: 每一个时步(time_step)输入到lstm单元的维度.(实际输入的数据size为[batch_size, input_size]) hidden_size: 确定了隐状态hidden_state的维度. 可以简单的看成: 构造了一个权重 , 隐含状态num_layers: 叠加的层数。如图所示num_layers为3batch_first: 输入数据的size为[batch_size, time_step,..
2021-10-08 23:05:55
663
原创 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
115
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人