Package ai.djl.ndarray
Class NDUtils
- java.lang.Object
-
- ai.djl.ndarray.NDUtils
-
public final class NDUtils extends java.lang.Object
A class containing utility methods for NDArray operations.
-
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static void
checkConcatInput(NDList list)
Check two criteria of concat input: 1.static Shape
getShapeFromEmptyNDArrayForReductionOp(Shape shape, int axis)
-