Table Of Contents
Table Of Contents

mx.symbol.broadcast_axes

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.

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_axes(...)

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.