Skip to contents

Primitive Transpose

Source: R/primitives.R
nvl_transpose.Rd

Permutes the dimensions of a tensor.

Usage

nvl_transpose(operand, permutation)

Arguments

operand

(tensorish)
Tensorish value of any data type.

permutation

(integer())
Specifies the new ordering of dimensions. Must be a permutation of seq_len(ndims) where ndims is the number of dimensions of operand.

Value

tensorish
Has the same data type as the input and shape nv_shape(operand)[permutation]. It is ambiguous if the input is ambiguous.

Implemented Rules

  • stablehlo

  • backward

StableHLO

Lowers to stablehlo::hlo_transpose().

See also

nv_transpose(), t()

Examples

jit_eval({
  x <- nv_tensor(matrix(1:6, nrow = 2))
  nvl_transpose(x, permutation = c(2L, 1L))
})
#> AnvilTensor
#>  1 2
#>  3 4
#>  5 6
#> [ CPUi32{3,2} ]