torch.dot — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.dot. torch.dot(input, other, *, 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. input ( Tensor) – first tensor in the dot product, must be 1D.
torch.tensordot — PyTorch 1.10.1 documentation
pytorch.org › generated › torchLearn about PyTorch’s features and capabilities. Community. Join the PyTorch developer community to contribute, learn, and get your questions answered. Developer Resources. Find resources and get questions answered. Forums. A place to discuss PyTorch code, issues, install, research. Models (Beta) Discover, publish, and reuse pre-trained models