torch (version 0.0.2)

torch_mv: Mv

Description

Mv

Arguments

input

(Tensor) matrix to be multiplied

vec

(Tensor) vector to be multiplied

out

(Tensor, optional) the output tensor.

mv(input, vec, out=None) -> Tensor

Performs a matrix-vector product of the matrix input and the vector vec.

If input is a \((n \times m)\) tensor, vec is a 1-D tensor of size \(m\), out will be 1-D of size \(n\).

Examples

# NOT RUN {
if (torch_is_installed()) {

mat = torch_randn(c(2, 3))
vec = torch_randn(c(3))
torch_mv(mat, vec)
}
# }