mx.nd.size.array

Description

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

Example:

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

Arguments

Argument

Description

data

NDArray-or-Symbol.

Input Array.