I am using a linear layer like the following:
self.emb_to_seq = nn.Linear(512, 40000)
img_seq = self.emb_to_seq(V)
the tensor V has the shape of (8, 121, 512) and type of CudaFloatTensor.
When I want to print it as the following;
I get the following error:
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70
What is going on here?