io.improbable.keanu.tensor.ndj4

Class INDArrayShim

java.lang.Object

io.improbable.keanu.tensor.ndj4.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	gt(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	gte(org.nd4j.linalg.api.ndarray.INDArray mask, 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	lte(org.nd4j.linalg.api.ndarray.INDArray mask, 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	rdivi(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
static org.nd4j.linalg.api.ndarray.INDArray	rsubi(org.nd4j.linalg.api.ndarray.INDArray left, org.nd4j.linalg.api.ndarray.INDArray right)
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)

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)

rsubi

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

rdivi

public static org.nd4j.linalg.api.ndarray.INDArray rdivi(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)

gte

public static org.nd4j.linalg.api.ndarray.INDArray gte(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                       org.nd4j.linalg.api.ndarray.INDArray right)

lte

public static org.nd4j.linalg.api.ndarray.INDArray lte(org.nd4j.linalg.api.ndarray.INDArray mask,
                                                       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)