Skip to contents

Primitive Broadcast

Source: R/primitives.R
prim_broadcast_in_dim.Rd

Broadcasts an array to a new shape by replicating the data along new or size-1 dimensions.

Usage

prim_broadcast_in_dim(operand, shape, broadcast_dimensions)

Arguments

operand

(arrayish)
Arrayish value of any data type.

shape

(integer())
Target shape. Each mapped dimension must either match the corresponding operand dimension or the operand dimension must be 1.

broadcast_dimensions

(integer())
Maps each dimension of operand to a dimension of the output. Must have length equal to the number of dimensions of operand.

Value

arrayish
Has the same data type as the input and the given shape. It is ambiguous if the input is ambiguous.

Implemented Rules

  • stablehlo

  • quickr

  • reverse

StableHLO

Lowers to stablehlo::hlo_broadcast_in_dim().

See also

nv_broadcast_to()

Examples

x <- nv_array(c(1, 2, 3))
prim_broadcast_in_dim(x, shape = c(2, 3), broadcast_dimensions = 2L)
#> AnvlArray
#>  1 2 3
#>  1 2 3
#> [ CPUf32{2,3} ]