org.nd4j.linalg.cache

Class TadDescriptor

java.lang.Object

org.nd4j.linalg.cache.TadDescriptor

public class TadDescriptor
extends Object

This is utility class, made to compare TADs for caching purposes. Idea: for any given INDArray with any specific shape, TAD for specific dimension will always be the same. So it can be reused as much as we want. Of note here is that when used as a key, we preserve immutability of the shape buffer in the ndarray by copying the values with dataBufferToArray(DataBuffer)

Author:

[email protected]

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