Table Of Contents
Table Of Contents

mx.nd.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.

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.]]]

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.