torch.inner
torch.inner(input, other, *, out=None) -> Tensor
Computes the dot product for 1D tensors. For higher dimensions,
   sums the product of elements from "input" and "other" along their
   last dimension.
Note:
 If either "input" or "other" is a scalar, the result is
 equivalent to *torch.mul(input, other)*.If both "input" and
 "other" are non-scalars, the size of their last dimension must
 match and the result is equivalent to *torch.tensordot(input,
 other, dims=([-1], [-1]))*

Parameters:
      * input (Tensor) -- First input tensor
  * **other** (*Tensor*) -- Second input tensor

Keyword Arguments:
      out (Tensor, optional) -- Optional output tensor to
      write result into. The output shape is input.shape[:-1] +
      other.shape[:-1].
Example:
  # Dot product
  >>> torch.inner(torch.tensor([1, 2, 3]), torch.tensor([0, 2, 1]))
  tensor(7)

  # Multidimensional input tensors
