mx.symbol.shape_array

Description

Returns a 1D int64 array containing the shape of data.

Example:

shape_array([[1,2,3,4], [5,6,7,8]]) = [2,4]

Usage

mx.symbol.shape_array(...)

Arguments

Argument

Description

data

NDArray-or-Symbol.

Input Array.

name

string, optional.

Name of the resulting symbol.