org.nd4j.linalg.cache

Class TadDescriptor

java.lang.Object

org.nd4j.linalg.cache.TadDescriptor

public class TadDescriptor
extends Object

Constructor Summary

Constructors

Constructor and Description
TadDescriptor(INDArray array, int[] dimension) Pass in an ndarray to get the databuffer and the appropriate dimensions

Method Summary

Modifier and Type	Method and Description
static long[]	dataBufferToArray(DataBuffer buffer) Obtain the values from the shape buffer for the array

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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 from

dimension - 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