Skip to contents

Matrix Multiplication

Source: R/api.R
nv_matmul.Rd

Matrix multiplication of two tensors.

Usage

nv_matmul(lhs, rhs)

Arguments

lhs

(nv_tensor)

rhs

(nv_tensor)

Value

nv_tensor

Shapes

  • lhs: (b1, ..., bk, m, n)

  • rhs: (b1, ..., bk, n, p)

  • output: (b1, ..., bk, m, p)

Broadcasting

All dimensions but the last two are broadcasted.