On this page
torch.mv
torch.mv(input, vec, *, out=None) → Tensor
-
Performs a matrix-vector product of the matrix
input
and the vectorvec
.If
input
is a tensor,vec
is a 1-D tensor of size ,out
will be 1-D of size .Note
This function does not broadcast.
- Parameters
- Keyword Arguments
-
out (Tensor, optional) – the output tensor.
Example:
>>> mat = torch.randn(2, 3) >>> vec = torch.randn(3) >>> torch.mv(mat, vec) tensor([ 1.0404, -0.6361])
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.mv.html