1,问题描述
本人在pycharm中使用1维卷积,使用不了批处理,会报错,错误如下:
RuntimeError:Given groups = 1,weight of size [16,1,3],expected input [1,16,205] to have 1 channels,but got 16 channels instead
2,解决办法
当前数据形状为1行205列,即输入通道应为1,数据长度为205,批处理大小为16。
此时未转换之前的数据格式为16行205列,观察官方给出的1维卷积的对输入数据格式的要求,不难发现,要求是(批处理大小,输入数据通道数,数据长度),我们的数据(16,205),显然不满足要求。
即使用
input_data = torch.reshape(input_data, (16, 1, 205))
将数据转换为(16,1, 205)。即批处理为16,输入通道为1,数据长度为205。
竟然是这么简单的一个错误,还得学习啊,害。
感谢b站土堆up主,看他的神经网络课程时,听他说过这个方法:当现有数据和输入数据不一致时,通过reshape转换即可。
另:当有问题时不妨看看官方文档,说不定一看就明白了呢。