I changed this doc to reflect the fact that the axes cannot contain dim 0 (this is the batch dim), and the axes, when a scalar, excludes the first dim (the batch dim), so I think it best to replicate just this part. But I will put reference to tensordot() as a plus.