When I try to do a backward pass through a F.cosine_similarity(), the following error appears:
RuntimeError: invalid argument 1: the number of sizes provided must be greater or equal to the number of dimensions in the tensor at /pytorch/torch/lib/TH/generic/THTensor.c:299

I am running PyTorch version ‘0.3.0.post4’ (which I think is the latest one available as a pip package). The following piece of code should reproduce the error:
import torch
import torch.nn.functional as F
from torch.autograd import Variable

x1 = Variable(torch.FloatTensor([1, 2]), requires_grad=True)
x2 = Variable(torch.FloatTensor([3, 4]), requires_grad=True)
y = F.cosine_similarity(x1, x2, dim=0)
y.backward()

Is this a bug or am I doing something wrong? If it’s a bug, is it fixed in PyTorch master version?
Thanks in advance