There’s no tensordot in PyTorch. You can simulate it using a combination of exapand, * (mul), and sum.
Tensordot seems generally useful. If you’d like it, file an issue on the PyTorch github repo 119: