io.improbable.keanu.tensor

Class INDArrayShim

java.lang.Object

io.improbable.keanu.tensor.INDArrayShim

public class INDArrayShim
extends java.lang.Object

This class provides shim methods for the ND4J INDArray class. The INDArray broadcast operations are currently broken (https://github.com/deeplearning4j/deeplearning4j/issues/5893). Until this is fixed in the ND4J codebase, these methods can be used to work around the issue. The need for this should be reevaluated each time the ND4J dependency is updated.

To work around another issue in ND4J where you cannot broadcast a higher rank tensor onto a lower rank tensor, the shim broadcast operations ensure the higher rank tensor is always being operated on. In the case of subtract and minus, this requires a small change in the logic, as A - B != B - A and A / B != B / A.

Constructor Summary

Constructors

Constructor and Description
INDArrayShim()

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	addi(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	atan2(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	divi(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	eq(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	getGreaterThanMask(org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray right, org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)
static org.nd4j.linalg.api.ndarray.INDArray	getGreaterThanOrEqualToMask(org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray right, org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)
static org.nd4j.linalg.api.ndarray.INDArray	getLessThanMask(org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray right, org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)
static org.nd4j.linalg.api.ndarray.INDArray	getLessThanOrEqualToMask(org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray right, org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)
static org.nd4j.linalg.api.ndarray.INDArray	gt(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	lt(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	max(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	min(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	muli(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	pow(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	slice(org.nd4j.linalg.api.ndarray.INDArray tensor, int dimension, long index)
static void	startNewThreadForNd4j()
static org.nd4j.linalg.api.ndarray.INDArray	subi(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	sum(org.nd4j.linalg.api.ndarray.INDArray tensor, int... overDimensions)

Methods inherited from class java.lang.Object

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

Constructor Detail

INDArrayShim

public INDArrayShim()

Method Detail

startNewThreadForNd4j

public static void startNewThreadForNd4j()

muli

public static org.nd4j.linalg.api.ndarray.INDArray muli(org.nd4j.linalg.api.ndarray.INDArray left,
                                                        org.nd4j.linalg.api.ndarray.INDArray right)

divi

public static org.nd4j.linalg.api.ndarray.INDArray divi(org.nd4j.linalg.api.ndarray.INDArray left,
                                                        org.nd4j.linalg.api.ndarray.INDArray right)

addi

public static org.nd4j.linalg.api.ndarray.INDArray addi(org.nd4j.linalg.api.ndarray.INDArray left,
                                                        org.nd4j.linalg.api.ndarray.INDArray right)

subi

public static org.nd4j.linalg.api.ndarray.INDArray subi(org.nd4j.linalg.api.ndarray.INDArray left,
                                                        org.nd4j.linalg.api.ndarray.INDArray right)

pow

public static org.nd4j.linalg.api.ndarray.INDArray pow(org.nd4j.linalg.api.ndarray.INDArray left,
                                                       org.nd4j.linalg.api.ndarray.INDArray right)

max

public static org.nd4j.linalg.api.ndarray.INDArray max(org.nd4j.linalg.api.ndarray.INDArray left,
                                                       org.nd4j.linalg.api.ndarray.INDArray right)

min

public static org.nd4j.linalg.api.ndarray.INDArray min(org.nd4j.linalg.api.ndarray.INDArray left,
                                                       org.nd4j.linalg.api.ndarray.INDArray right)

atan2

public static org.nd4j.linalg.api.ndarray.INDArray atan2(org.nd4j.linalg.api.ndarray.INDArray left,
                                                         org.nd4j.linalg.api.ndarray.INDArray right)

lt

public static org.nd4j.linalg.api.ndarray.INDArray lt(org.nd4j.linalg.api.ndarray.INDArray left,
                                                      org.nd4j.linalg.api.ndarray.INDArray right)

gt

public static org.nd4j.linalg.api.ndarray.INDArray gt(org.nd4j.linalg.api.ndarray.INDArray left,
                                                      org.nd4j.linalg.api.ndarray.INDArray right)

eq

public static org.nd4j.linalg.api.ndarray.INDArray eq(org.nd4j.linalg.api.ndarray.INDArray left,
                                                      org.nd4j.linalg.api.ndarray.INDArray right)

getGreaterThanMask

public static org.nd4j.linalg.api.ndarray.INDArray getGreaterThanMask(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                      org.nd4j.linalg.api.ndarray.INDArray right,
                                                                      org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)

getGreaterThanOrEqualToMask

public static org.nd4j.linalg.api.ndarray.INDArray getGreaterThanOrEqualToMask(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                               org.nd4j.linalg.api.ndarray.INDArray right,
                                                                               org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)

getLessThanMask

public static org.nd4j.linalg.api.ndarray.INDArray getLessThanMask(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                   org.nd4j.linalg.api.ndarray.INDArray right,
                                                                   org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)

getLessThanOrEqualToMask

public static org.nd4j.linalg.api.ndarray.INDArray getLessThanOrEqualToMask(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                            org.nd4j.linalg.api.ndarray.INDArray right,
                                                                            org.nd4j.linalg.api.buffer.DataBuffer.Type bufferType)

sum

public static org.nd4j.linalg.api.ndarray.INDArray sum(org.nd4j.linalg.api.ndarray.INDArray tensor,
                                                       int... overDimensions)

slice

public static org.nd4j.linalg.api.ndarray.INDArray slice(org.nd4j.linalg.api.ndarray.INDArray tensor,
                                                         int dimension,
                                                         long index)