torch.dot
- torch.dot(input, tensor, *, out=None) Tensor
Computes the dot product of two 1D tensors.
Note
Unlike NumPy’s dot, torch.dot intentionally only supports computing the dot product of two 1D tensors with the same number of elements.
- Parameters
- Keyword Arguments
out (Tensor, optional) – the output tensor.
Example:
>>> torch.dot(torch.tensor([2, 3]), torch.tensor([2, 1])) tensor(7) >>> t1, t2 = torch.tensor([0, 1]), torch.tensor([2, 3]) >>> torch.dot(t1, t2) tensor(3)