io.improbable.keanu.tensor.intgr

Class Nd4jIntegerTensor

java.lang.Object

io.improbable.keanu.tensor.ndj4.Nd4jTensor<T,TENSOR>

io.improbable.keanu.tensor.ndj4.Nd4jNumberTensor<T,TENSOR>

io.improbable.keanu.tensor.ndj4.Nd4jFixedPointTensor<java.lang.Integer,IntegerTensor>

io.improbable.keanu.tensor.intgr.Nd4jIntegerTensor

All Implemented Interfaces:

io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>, io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>, FixedPointTensor<java.lang.Integer,IntegerTensor>, IntegerTensor, NumberTensor<java.lang.Integer,IntegerTensor>, Tensor<java.lang.Integer,IntegerTensor>

public class Nd4jIntegerTensor
extends Nd4jFixedPointTensor<java.lang.Integer,IntegerTensor>
implements IntegerTensor

Class for representing n-dimensional arrays of integers. This is backed by Nd4j.

Nested Class Summary

Nested classes/interfaces inherited from interface io.improbable.keanu.tensor.Tensor

Tensor.FlattenedView<N>

Field Summary

Fields inherited from class io.improbable.keanu.tensor.ndj4.Nd4jTensor

tensor

Fields inherited from interface io.improbable.keanu.tensor.Tensor

ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE

Constructor Summary

Constructors

Constructor and Description
Nd4jIntegerTensor(org.nd4j.linalg.api.ndarray.INDArray tensor)
Nd4jIntegerTensor(int[] data, long[] shape)

Method Summary

Modifier and Type	Method and Description
static Nd4jIntegerTensor	arange(int start, int end)
static Nd4jDoubleTensor	arange(int start, int end, int stepSize)
java.lang.Integer[]	asFlatArray()
double[]	asFlatDoubleArray()
int[]	asFlatIntegerArray()
protected IntegerTensor	create(org.nd4j.linalg.api.ndarray.INDArray tensor)
static Nd4jIntegerTensor	create(int[] values, long[] shape)
static Nd4jIntegerTensor	create(int value, long[] shape)
boolean	equalsWithinEpsilon(IntegerTensor o, java.lang.Integer epsilon)
static Nd4jIntegerTensor	eye(long n)
Tensor.FlattenedView<java.lang.Integer>	getFlattenedView()
protected java.lang.Integer	getNumber(java.lang.Number number)
protected org.nd4j.linalg.api.ndarray.INDArray	getTensor(Tensor<java.lang.Integer,?> tensor)
protected IntegerTensor	getThis()
IntegerTensor	greaterThanMask(IntegerTensor greaterThanThis)
IntegerTensor	greaterThanOrEqualToMask(IntegerTensor greaterThanOrEqualToThis)
IntegerTensor	lessThanMask(IntegerTensor lessThanThis)
IntegerTensor	lessThanOrEqualToMask(IntegerTensor lessThanOrEqualToThis)
static Nd4jIntegerTensor	ones(long[] shape)
IntegerTensor	safeLogTimesInPlace(IntegerTensor y)
static Nd4jIntegerTensor	scalar(int scalarValue)
protected IntegerTensor	set(org.nd4j.linalg.api.ndarray.INDArray tensor)
IntegerTensor	setAllInPlace(java.lang.Integer value)
DoubleTensor	toDouble()
IntegerTensor	toInteger()
static Nd4jIntegerTensor	zeros(long[] shape)

Methods inherited from class io.improbable.keanu.tensor.ndj4.Nd4jFixedPointTensor

modInPlace, modInPlace

Methods inherited from class io.improbable.keanu.tensor.ndj4.Nd4jNumberTensor

absInPlace, applyInPlace, argMax, argMax, argMin, argMin, average, clampInPlace, cumProdInPlace, cumSumInPlace, div, divInPlace, divInPlace, elementwiseEquals, elementwiseEquals, fromMask, greaterThan, greaterThan, greaterThanOrEqual, greaterThanOrEqual, lessThan, lessThan, lessThanOrEqual, lessThanOrEqual, matrixMultiply, max, maxInPlace, min, minInPlace, minus, minusInPlace, minusInPlace, plus, plusInPlace, plusInPlace, powInPlace, powInPlace, product, product, reverseDivInPlace, reverseDivInPlace, reverseMinusInPlace, reverseMinusInPlace, setWithMaskInPlace, standardDeviation, sum, sum, tensorMultiply, times, timesInPlace, timesInPlace, unaryMinusInPlace

Methods inherited from class io.improbable.keanu.tensor.ndj4.Nd4jTensor

broadcast, diag, duplicate, equals, get, getLength, getRank, getShape, getStride, getTensor, hashCode, permute, reshape, slice, slice, split, take, toString, transpose

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface io.improbable.keanu.tensor.intgr.IntegerTensor

concat, concat, create, create, create, div, eye, max, min, minus, plus, pow, reverseDiv, reverseMinus, stack, times, vector

Methods inherited from interface io.improbable.keanu.tensor.FixedPointTensor

mod, mod, modInPlace, modInPlace

Methods inherited from interface io.improbable.keanu.tensor.NumberTensor

abs, absInPlace, apply, applyInPlace, argMax, argMax, argMin, argMin, average, clamp, clampInPlace, cumProd, cumProdInPlace, cumSum, cumSumInPlace, div, div, divInPlace, divInPlace, greaterThan, greaterThan, greaterThanOrEqual, greaterThanOrEqual, lessThan, lessThan, lessThanOrEqual, lessThanOrEqual, matrixMultiply, max, max, maxInPlace, min, min, minInPlace, minus, minus, minusInPlace, minusInPlace, plus, plus, plusInPlace, plusInPlace, pow, pow, powInPlace, powInPlace, product, product, reverseDiv, reverseDiv, reverseDivInPlace, reverseDivInPlace, reverseMinus, reverseMinus, reverseMinusInPlace, reverseMinusInPlace, safeLogTimes, setWithMask, setWithMaskInPlace, standardDeviation, sum, sum, tensorMultiply, times, times, timesInPlace, timesInPlace, unaryMinus, unaryMinusInPlace

Methods inherited from interface io.improbable.keanu.tensor.Tensor

asFlatList, broadcast, create, createFilled, diag, duplicate, elementwiseEquals, elementwiseEquals, elementwiseEquals, expandDims, get, getLength, getRank, getShape, getStride, getValue, hasSameShapeAs, hasSameShapeAs, isLengthOne, isMatrix, isScalar, isVector, moveAxis, permute, reshape, scalar, setValue, slice, slice, slice, sliceAlongDimension, split, squeeze, swapAxis, take, transpose

Constructor Detail

Nd4jIntegerTensor

public Nd4jIntegerTensor(int[] data,
                         long[] shape)

Nd4jIntegerTensor

public Nd4jIntegerTensor(org.nd4j.linalg.api.ndarray.INDArray tensor)

Method Detail

getTensor

protected org.nd4j.linalg.api.ndarray.INDArray getTensor(Tensor<java.lang.Integer,?> tensor)

Specified by:

getTensor in class Nd4jTensor<java.lang.Integer,IntegerTensor>

scalar

public static Nd4jIntegerTensor scalar(int scalarValue)

Specified by:

scalar in interface IntegerTensor

Specified by:

scalar in interface Tensor<java.lang.Integer,IntegerTensor>

create

public static Nd4jIntegerTensor create(int[] values,
                                       long[] shape)

Specified by:

create in interface IntegerTensor

create

public static Nd4jIntegerTensor create(int value,
                                       long[] shape)

Specified by:

create in interface IntegerTensor

ones

public static Nd4jIntegerTensor ones(long[] shape)

Specified by:

ones in interface IntegerTensor

eye

public static Nd4jIntegerTensor eye(long n)

zeros

public static Nd4jIntegerTensor zeros(long[] shape)

Specified by:

zeros in interface IntegerTensor

arange

public static Nd4jIntegerTensor arange(int start,
                                       int end)

arange

public static Nd4jDoubleTensor arange(int start,
                                      int end,
                                      int stepSize)

equalsWithinEpsilon

public boolean equalsWithinEpsilon(IntegerTensor o,
                                   java.lang.Integer epsilon)

Specified by:

equalsWithinEpsilon in interface NumberTensor<java.lang.Integer,IntegerTensor>

getNumber

protected java.lang.Integer getNumber(java.lang.Number number)

Specified by:

getNumber in class Nd4jNumberTensor<java.lang.Integer,IntegerTensor>

setAllInPlace

public IntegerTensor setAllInPlace(java.lang.Integer value)

Specified by:

setAllInPlace in interface NumberTensor<java.lang.Integer,IntegerTensor>

safeLogTimesInPlace

public IntegerTensor safeLogTimesInPlace(IntegerTensor y)

Specified by:

safeLogTimesInPlace in interface NumberTensor<java.lang.Integer,IntegerTensor>

greaterThanMask

public IntegerTensor greaterThanMask(IntegerTensor greaterThanThis)

Specified by:

greaterThanMask in interface NumberTensor<java.lang.Integer,IntegerTensor>

greaterThanOrEqualToMask

public IntegerTensor greaterThanOrEqualToMask(IntegerTensor greaterThanOrEqualToThis)

Specified by:

greaterThanOrEqualToMask in interface NumberTensor<java.lang.Integer,IntegerTensor>

lessThanMask

public IntegerTensor lessThanMask(IntegerTensor lessThanThis)

Specified by:

lessThanMask in interface NumberTensor<java.lang.Integer,IntegerTensor>

lessThanOrEqualToMask

public IntegerTensor lessThanOrEqualToMask(IntegerTensor lessThanOrEqualToThis)

Specified by:

lessThanOrEqualToMask in interface NumberTensor<java.lang.Integer,IntegerTensor>

toDouble

public DoubleTensor toDouble()

Specified by:

toDouble in interface NumberTensor<java.lang.Integer,IntegerTensor>

toInteger

public IntegerTensor toInteger()

Specified by:

toInteger in interface NumberTensor<java.lang.Integer,IntegerTensor>

create

protected IntegerTensor create(org.nd4j.linalg.api.ndarray.INDArray tensor)

Specified by:

create in class Nd4jTensor<java.lang.Integer,IntegerTensor>

set

protected IntegerTensor set(org.nd4j.linalg.api.ndarray.INDArray tensor)

Specified by:

set in class Nd4jTensor<java.lang.Integer,IntegerTensor>

getThis

protected IntegerTensor getThis()

Specified by:

getThis in class Nd4jTensor<java.lang.Integer,IntegerTensor>

asFlatDoubleArray

public double[] asFlatDoubleArray()

Specified by:

asFlatDoubleArray in interface NumberTensor<java.lang.Integer,IntegerTensor>

asFlatIntegerArray

public int[] asFlatIntegerArray()

Specified by:

asFlatIntegerArray in interface NumberTensor<java.lang.Integer,IntegerTensor>

asFlatArray

public java.lang.Integer[] asFlatArray()

Specified by:

asFlatArray in interface Tensor<java.lang.Integer,IntegerTensor>

getFlattenedView

public Tensor.FlattenedView<java.lang.Integer> getFlattenedView()

Specified by:

getFlattenedView in interface Tensor<java.lang.Integer,IntegerTensor>