Table Of Contents
Table Of Contents

mxnet.ndarray.size_array

mxnet.ndarray.size_array(data=None, out=None, name=None, **kwargs)

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

Example:

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

Defined in src/operator/tensor/elemwise_unary_op_basic.cc:L572

Parameters

  • data (NDArray) – Input Array.

  • out (NDArray, optional) – The output NDArray to hold the result.

Returns

out – The output of this function.

Return type

NDArray or list of NDArrays

Previous
mxnet.ndarray.shape_array
Next
mxnet.ndarray.broadcast_axes