Skip to contents

Matrix Multiplication

Source: R/api.R
nv_matmul.Rd

Matrix multiplication of two arrays. You can also use the %*% operator. Supports batched matrix multiplication when inputs have more than 2 axes.

Usage

nv_matmul(lhs, rhs, precision = "highest")

Arguments

lhs, rhs

(arrayish)
Arrays with at least 2 axes. Operands are promoted to a common data type.

precision

(character(1))
Controls the trade-off between speed and numerical accuracy of the operation. One of "highest" (default), "high" or "default". See prim_dot_general() for details.

Value

arrayish

Shapes

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

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

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

See also

prim_dot_general() for the underlying primitive.

Examples

x <- nv_matrix(1:6, nrow = 2)
y <- nv_matrix(1:6, nrow = 3)
x %*% y
#> AnvlArray
#>  22 49
#>  28 64
#> [ CPUi32{2,2} ]