io.improbable.keanu.tensor.ndj4

Class Nd4jFloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

java.lang.Object

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

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

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

All Implemented Interfaces:

io.improbable.keanu.kotlin.DoubleOperators<TENSOR>, io.improbable.keanu.kotlin.NumberOperators<TENSOR>, FloatingPointTensor<T,TENSOR>, NumberTensor<T,TENSOR>, Tensor<T,TENSOR>

Direct Known Subclasses:

Nd4jDoubleTensor

public abstract class Nd4jFloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>
extends Nd4jNumberTensor<T,TENSOR>
implements FloatingPointTensor<T,TENSOR>

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
Nd4jFloatingPointTensor(org.nd4j.linalg.api.ndarray.INDArray tensor)

Method Summary

Modifier and Type	Method and Description
TENSOR	acoshInPlace()
TENSOR	acosInPlace()
TENSOR	asinhInPlace()
TENSOR	asinInPlace()
TENSOR	atan2InPlace(T y)
TENSOR	atan2InPlace(TENSOR y)
TENSOR	atanhInPlace()
TENSOR	atanInPlace()
TENSOR	ceilInPlace()
TENSOR	choleskyDecomposition()
TENSOR	clampInPlace(TENSOR min, TENSOR max)
TENSOR	coshInPlace()
TENSOR	cosInPlace()
T	determinant()
TENSOR	exp2InPlace()
TENSOR	expInPlace()
TENSOR	expM1InPlace()
TENSOR	floorInPlace()
TENSOR	log10InPlace()
TENSOR	log1pInPlace()
TENSOR	log2InPlace()
TENSOR	logInPlace()
TENSOR	matrixInverse()
TENSOR	matrixMultiply(TENSOR value)
BooleanTensor	notNaN()
T	product()
TENSOR	reciprocalInPlace()
TENSOR	roundInPlace()
TENSOR	setAllInPlace(T value)
TENSOR	sigmoidInPlace()
TENSOR	sinhInPlace()
TENSOR	sinInPlace()
TENSOR	sqrtInPlace()
TENSOR	standardizeInPlace()
TENSOR	tanhInPlace()
TENSOR	tanInPlace()
TENSOR	tensorMultiply(TENSOR value, int[] dimsLeft, int[] dimsRight)

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

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

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

broadcast, create, diag, duplicate, equals, get, getLength, getRank, getShape, getStride, getTensor, getTensor, getThis, hashCode, permute, reshape, set, 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.FloatingPointTensor

acos, acosh, asin, asinh, atan, atan2, atan2, atanh, ceil, cos, cosh, digamma, digammaInPlace, exp, exp2, expM1, floor, isFinite, isInfinite, isNaN, isNegativeInfinity, isPositiveInfinity, log, log10, log1p, log2, logAddExp, logAddExp2, logAddExp2InPlace, logAddExpInPlace, logGamma, logGammaInPlace, nanArgMax, nanArgMax, nanArgMin, nanArgMin, reciprocal, replaceNaN, replaceNaNInPlace, round, sigmoid, sin, sinh, sqrt, standardize, tan, tanh

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

abs, absInPlace, apply, applyInPlace, argMax, argMax, argMin, argMin, asFlatDoubleArray, asFlatIntegerArray, average, clamp, cumProd, cumProdInPlace, cumSum, cumSumInPlace, div, div, divInPlace, divInPlace, equalsWithinEpsilon, greaterThan, greaterThan, greaterThanMask, greaterThanOrEqual, greaterThanOrEqual, greaterThanOrEqualToMask, lessThan, lessThan, lessThanMask, lessThanOrEqual, lessThanOrEqual, lessThanOrEqualToMask, max, max, maxInPlace, min, min, minInPlace, minus, minus, minusInPlace, minusInPlace, plus, plus, plusInPlace, plusInPlace, pow, pow, powInPlace, powInPlace, product, reverseDiv, reverseDiv, reverseDivInPlace, reverseDivInPlace, reverseMinus, reverseMinus, reverseMinusInPlace, reverseMinusInPlace, safeLogTimes, safeLogTimesInPlace, setWithMask, setWithMaskInPlace, standardDeviation, sum, sum, times, times, timesInPlace, timesInPlace, toDouble, toInteger, unaryMinus, unaryMinusInPlace

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

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

Methods inherited from interface io.improbable.keanu.kotlin.DoubleOperators

div, minus, plus, pow, reverseDiv, reverseMinus, times

Constructor Detail

Nd4jFloatingPointTensor

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

Method Detail

matrixInverse

public TENSOR matrixInverse()

Specified by:

matrixInverse in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

choleskyDecomposition

public TENSOR choleskyDecomposition()

Specified by:

choleskyDecomposition in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

matrixMultiply

public TENSOR matrixMultiply(TENSOR value)

Specified by:

matrixMultiply in interface NumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

Overrides:

matrixMultiply in class Nd4jNumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

tensorMultiply

public TENSOR tensorMultiply(TENSOR value,
                             int[] dimsLeft,
                             int[] dimsRight)

Specified by:

tensorMultiply in interface NumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

Overrides:

tensorMultiply in class Nd4jNumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

sqrtInPlace

public TENSOR sqrtInPlace()

Specified by:

sqrtInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

logInPlace

public TENSOR logInPlace()

Specified by:

logInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

reciprocalInPlace

public TENSOR reciprocalInPlace()

Specified by:

reciprocalInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

sinInPlace

public TENSOR sinInPlace()

Specified by:

sinInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

cosInPlace

public TENSOR cosInPlace()

Specified by:

cosInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

tanInPlace

public TENSOR tanInPlace()

Specified by:

tanInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

atanInPlace

public TENSOR atanInPlace()

Specified by:

atanInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

atan2InPlace

public TENSOR atan2InPlace(T y)

Specified by:

atan2InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

atan2InPlace

public TENSOR atan2InPlace(TENSOR y)

Specified by:

atan2InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

asinInPlace

public TENSOR asinInPlace()

Specified by:

asinInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

acosInPlace

public TENSOR acosInPlace()

Specified by:

acosInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

sinhInPlace

public TENSOR sinhInPlace()

Specified by:

sinhInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

coshInPlace

public TENSOR coshInPlace()

Specified by:

coshInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

tanhInPlace

public TENSOR tanhInPlace()

Specified by:

tanhInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

asinhInPlace

public TENSOR asinhInPlace()

Specified by:

asinhInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

acoshInPlace

public TENSOR acoshInPlace()

Specified by:

acoshInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

atanhInPlace

public TENSOR atanhInPlace()

Specified by:

atanhInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

expInPlace

public TENSOR expInPlace()

Specified by:

expInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

log1pInPlace

public TENSOR log1pInPlace()

Specified by:

log1pInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

log2InPlace

public TENSOR log2InPlace()

Specified by:

log2InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

log10InPlace

public TENSOR log10InPlace()

Specified by:

log10InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

exp2InPlace

public TENSOR exp2InPlace()

Specified by:

exp2InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

expM1InPlace

public TENSOR expM1InPlace()

Specified by:

expM1InPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

standardizeInPlace

public TENSOR standardizeInPlace()

Specified by:

standardizeInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

setAllInPlace

public TENSOR setAllInPlace(T value)

Specified by:

setAllInPlace in interface NumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

clampInPlace

public TENSOR clampInPlace(TENSOR min,
                           TENSOR max)

Specified by:

clampInPlace in interface NumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

Overrides:

clampInPlace in class Nd4jNumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

ceilInPlace

public TENSOR ceilInPlace()

Specified by:

ceilInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

floorInPlace

public TENSOR floorInPlace()

Specified by:

floorInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

roundInPlace

public TENSOR roundInPlace()

Specified by:

roundInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

sigmoidInPlace

public TENSOR sigmoidInPlace()

Specified by:

sigmoidInPlace in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

determinant

public T determinant()

Specified by:

determinant in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

product

public T product()

Specified by:

product in interface NumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

Overrides:

product in class Nd4jNumberTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>

notNaN

public BooleanTensor notNaN()

Specified by:

notNaN in interface FloatingPointTensor<T extends java.lang.Number,TENSOR extends FloatingPointTensor<T,TENSOR>>