mx.symbol.expand_dims

Description

Inserts a new axis of size 1 into the array shape For example, given x with shape (2,3,4), then expand_dims(x, axis=1) will return a new array with shape (2,1,3,4).

Usage

mx.symbol.expand_dims(...)

Arguments

Argument

Description

data

NDArray-or-Symbol.

Source input

axis

int, required.

Position where new axis is to be inserted. Suppose that the input NDArray’s dimension is ndim, the range of the inserted axis is [-ndim, ndim]

name

string, optional.

Name of the resulting symbol.