Does torch.diag work on variables without calling .data ? I’m getting weird behavior:
x = Variable(torch.ones((5,5)))
print(torch.diag(x)) # prints diag
print(torch.diag(x,1)) # still prints middle diag
print(torch.diag(x,2)) # still prints middle diag
print(torch.diag(x.data, 2)) # prints correct diag

EDIT: this was a small bug which should be fixed now.