torch_mv: Mv

View source: R/gen-namespace.R

torch_mvR Documentation

Mv

Description

Mv

Usage

torch_mv(self, vec)

Arguments

self

(Tensor) matrix to be multiplied

vec

(Tensor) vector to be multiplied

mv(input, vec, out=NULL) -> 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.

Note

This function does not broadcast .

Examples

if (torch_is_installed()) {

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

torch documentation built on May 29, 2024, 9:54 a.m.