- 博客(2)
- 收藏
- 关注
原创 CNN和LSTM的输入数据维度
CNN:对于torch.nn.Conv2d()来说,输入的数据需要是4维的,即[batch_size,channel,image_x,iamge_y]如果每次输入一个图像的话通道就是1pg.CNN的第一层是torch.nn.Conv2d(1,64)说明输出通道是64LSTM:(seq_len, batch, input_size)其中:seq_len表示的是句子的长度batch表示的是一次往LSTM中输入句子的数目input_size表示的是输入的维度
2022-12-16 14:33:17
2564
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人