Tensordot lets you specify contraction axes.
Einsum is foolproof in the way that you specify the dimension of each tensor.
I guess PyTorch is more modelled after NumPy than anything else, though.
Best regards
Thomas