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 |
---|---|
|
NDArray-or-Symbol. Input Array. |
Value¶
out
The result mx.ndarray
Link to Source Code: http://github.com/apache/incubator-mxnet/blob/1.6.0/src/operator/tensor/elemwise_unary_op_basic.cc#L574