Hello! I have a 1D tensor, with 500 entries and I want to pass it to a conv1d. After I put it into a mini batch of size 10 and I call x.size() I get this output: torch.Size([10, 500]). However the conv1d gives an error, requiring a 3D tensor, so I tried to add one more dimension, corresponding to th…

Passing a 1d tensor to conv1d

InnovArul (Arul) February 17, 2019, 5:49am 2

I guess you are using pytorch <= 0.4. Try to wrap the tensors with autograd.Variable. from torch.autograd import Variable input = Variable(input)

show post in topic

Powered by Discourse, best viewed with JavaScript enabled