org.nd4j.linalg.util

Class LinearUtil

java.lang.Object

org.nd4j.linalg.util.LinearUtil

public class LinearUtil
extends Object

Util class for handling edge cases where linear view is desirable but not usable (gpus)

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
LinearUtil()

Method Summary

Modifier and Type	Method and Description
static int	linearStride(INDArray arr) When needing to look at an array that's a matrix or some other tensor we typically need to map those arrays element wise (take axpy or uploading data as a linear buffer for example) This handles finding the proper offset for the situations where a >= 2d array needs to be interpreted in a linear form that may not be used outside of those contexts.

Methods inherited from class java.lang.Object

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

Constructor Detail

LinearUtil

public LinearUtil()

Method Detail

linearStride

public static int linearStride(INDArray arr)

When needing to look at an array that's a matrix or some other tensor we typically need to map those arrays element wise (take axpy or uploading data as a linear buffer for example) This handles finding the proper offset for the situations where a >= 2d array needs to be interpreted in a linear form that may not be used outside of those contexts. Note that this is different than linear view in that while this does technically interpret the buffer as a linear element wise buffer, this does not preserve structure which maybe needed for some operations. This is purely for operations like uploading to the gpu or element wise operations like axpy

Parameters:

arr - the array to get the stride for

Returns:

the linear stride (USED ONLY FOR AXPY AND ELEMENT WISE OPERATIONS WHERE THE BUFFER IS DIRECTLY INVOLVED) for this ndarray