I wrote a simple sinnpet on torch.diag, but it throws error, what’s the difference between torch.cuda.DoubleTensor
and torch.cuda.DoubleTensor source
?
Input is not a keyward arg! Try removing input=
.
1 Like
Got it, it makes sense, Thanks!