io.improbable.keanu.tensor.jvm

Class JVMNumberTensor<T extends java.lang.Number,TENSOR extends NumberTensor<T,TENSOR>,B extends PrimitiveNumberWrapper<T,B>>

java.lang.Object

io.improbable.keanu.tensor.jvm.JVMTensor<T,TENSOR,B>

io.improbable.keanu.tensor.jvm.JVMNumberTensor<T,TENSOR,B>

All Implemented Interfaces:

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

Direct Known Subclasses:

JVMFloatingPointTensor

public abstract class JVMNumberTensor<T extends java.lang.Number,TENSOR extends NumberTensor<T,TENSOR>,B extends PrimitiveNumberWrapper<T,B>>
extends JVMTensor<T,TENSOR,B>
implements NumberTensor<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.jvm.JVMTensor

buffer, shape, stride

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

ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	JVMNumberTensor(B buffer, long[] shape, long[] stride)

Method Summary

Modifier and Type	Method and Description
TENSOR	cumProdInPlace(int requestedDimension)
TENSOR	cumSumInPlace(int requestedDimension)
protected abstract JVMBuffer.PrimitiveNumberWrapperFactory<T,B>	getFactory()
T	product()
TENSOR	product(int... overDimensions)
T	sum()
TENSOR	sum(int... overDimensions)

Methods inherited from class io.improbable.keanu.tensor.jvm.JVMTensor

argCompare, argCompare, argCompare, argCompare, broadcast, broadcastableBinaryOpWithAutoBroadcast, concat, concatOnDimensionZero, create, diag, diag, equals, get, getLength, getRank, getShape, getStride, hashCode, isApply, permute, permute, reshape, set, slice, slice, slice, split, split, toString

Methods inherited from class java.lang.Object

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

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

abs, absInPlace, apply, applyInPlace, argMax, argMax, argMin, argMin, asFlatDoubleArray, asFlatIntegerArray, average, clamp, clampInPlace, cumProd, cumSum, div, div, divInPlace, divInPlace, equalsWithinEpsilon, greaterThan, greaterThan, greaterThanMask, greaterThanOrEqual, greaterThanOrEqual, greaterThanOrEqualToMask, lessThan, lessThan, lessThanMask, lessThanOrEqual, lessThanOrEqual, lessThanOrEqualToMask, matrixMultiply, max, max, maxInPlace, min, min, minInPlace, minus, minus, minusInPlace, minusInPlace, plus, plus, plusInPlace, plusInPlace, pow, pow, powInPlace, powInPlace, reverseDiv, reverseDiv, reverseDivInPlace, reverseDivInPlace, reverseMinus, reverseMinus, reverseMinusInPlace, reverseMinusInPlace, safeLogTimes, safeLogTimesInPlace, setAllInPlace, setWithMask, setWithMaskInPlace, standardDeviation, tensorMultiply, 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

Constructor Detail

JVMNumberTensor

protected JVMNumberTensor(B buffer,
                          long[] shape,
                          long[] stride)

Method Detail

sum

public T sum()

Specified by:

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

product

public T product()

Specified by:

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

sum

public TENSOR sum(int... overDimensions)

Specified by:

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

product

public TENSOR product(int... overDimensions)

Specified by:

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

cumSumInPlace

public TENSOR cumSumInPlace(int requestedDimension)

Specified by:

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

cumProdInPlace

public TENSOR cumProdInPlace(int requestedDimension)

Specified by:

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

getFactory

protected abstract JVMBuffer.PrimitiveNumberWrapperFactory<T,B> getFactory()

Specified by:

getFactory in class JVMTensor<T extends java.lang.Number,TENSOR extends NumberTensor<T,TENSOR>,B extends PrimitiveNumberWrapper<T,B>>