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

Arguments

Argument

Description

data

NDArray-or-Symbol.

Input Array.