Package org.nd4j.linalg.cache
Class TadDescriptor
- java.lang.Object
-
- org.nd4j.linalg.cache.TadDescriptor
-
public class TadDescriptor extends Object
-
-
Constructor Summary
Constructors Constructor Description TadDescriptor(INDArray array, int[] dimension)
Pass in an ndarray to get the databuffer and the appropriate dimensions
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static long[]
dataBufferToArray(DataBuffer buffer)
Obtain the values from the shape buffer for the array
-
-
-
Constructor Detail
-
TadDescriptor
public TadDescriptor(INDArray array, int[] dimension)
Pass in an ndarray to get the databuffer and the appropriate dimensions- Parameters:
array
- the array to pass in to get the shape info fromdimension
- the dimensions for the TAD
-
-
Method Detail
-
dataBufferToArray
public static long[] dataBufferToArray(DataBuffer buffer)
Obtain the values from the shape buffer for the array- Parameters:
buffer
- the buffer to get the values from- Returns:
- the int array version of this data buffer
-
-