Method size

api/src/main/java/ai/djl/ndarray/NDArray.java:215–215

Returns the size of this NDArray along a given axis. @param axis the axis to return the size for @return the size of this NDArray along a given axis

(int axis)

     * @return the size of this {@code NDArray} along a given axis
     */
    default long size(int axis) {
        return getShape().size(axis);
    }