RuntimeError: expand(torch.FloatTensor{[1, 512, 1, 1]}, size=[512]): the number of sizes provided (1) must be greater or equal to the number of dimensions in the tensor (4)

[jasonachonu]

when I ran the code i got a mismatched error

RuntimeError: expand(torch.FloatTensor{[1, 512, 1, 1]}, size=[512]): the number of sizes provided (1) must be greater or equal to the number of dimensions in the tensor (4)

can you please guide me on how to resolve this

[Jhhuangkay]

@jasonachonu

You can change the

original code: my_embedding = torch.zeros(512)

to: my_embedding = torch.zeros(1, 512, 1, 1)

Then, your problem will be solved.