Oke, the input to the 1D Convolutional layer must be of shape batch size x channel x spatial_size. In your case, it’s only channel x spatial_size. This can be fixed by model(data_pixel2[i].unsqueeze(0)) which adds another dimension in the tensor to be of shape (1,1,94)