a = torch.tensor((1, 2, 3))
b = torch.tensor((4, 5, 6))
torch.tensordot(a, b, dims=0)
: RuntimeError: unsupported input to tensordot, got dims=0
np.tensordot(a, b, axes=0)
: array([[ 4, 5, 6],
[ 8, 10, 12],
[12, 15, 18]])
There may be other inconsistencies.
The documentation and examples for torch.tensordot could also use an update. If updating the docs it'd be nice to also add a seealso section and close #50802.
Sample program:
There may be other inconsistencies.
The documentation and examples for torch.tensordot could also use an update. If updating the docs it'd be nice to also add a seealso section and close #50802.
cc @jianyuh @nikitaved @pearu @mruberry @walterddr @IvanYashchuk @xwang233 @lezcano