mx.symbol.broadcast_axis

Description

Broadcasts the input array over particular axes.

Broadcasting is allowed on axes with size 1, such as from (2,1,3,1) to (2,8,3,9). Elements will be duplicated on the broadcasted axes.

broadcast_axes is an alias to the function broadcast_axis.

Example:

// given x of shape (1,2,1)
x = [[[ 1.],
[ 2.]]]

// broadcast x on on axis 2
broadcast_axis(x, axis=2, size=3) = [[[ 1.,  1.,  1.],
[ 2.,  2.,  2.]]]
// broadcast x on on axes 0 and 2
broadcast_axis(x, axis=(0,2), size=(2,3)) = [[[ 1.,  1.,  1.],
[ 2.,  2.,  2.]],
[[ 1.,  1.,  1.],
[ 2.,  2.,  2.]]]

Usage

mx.symbol.broadcast_axis(...)

Arguments

Argument

Description

data

NDArray-or-Symbol.

The input

axis

Shape(tuple), optional, default=[].

The axes to perform the broadcasting.

size

Shape(tuple), optional, default=[].

Target sizes of the broadcasting axes.

name

string, optional.

Name of the resulting symbol.