io.improbable.keanu.tensor.intgr

Class JVMIntegerTensor

java.lang.Object

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

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

io.improbable.keanu.tensor.jvm.JVMFixedPointTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

io.improbable.keanu.tensor.intgr.JVMIntegerTensor

All Implemented Interfaces:

BaseFixedPointTensor<BooleanTensor,IntegerTensor,DoubleTensor,java.lang.Integer,IntegerTensor>, BaseNumberTensor<BooleanTensor,IntegerTensor,DoubleTensor,java.lang.Integer,IntegerTensor>, BaseTensor<BooleanTensor,java.lang.Integer,IntegerTensor>, 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 JVMIntegerTensor
extends JVMFixedPointTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>
implements IntegerTensor

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.BaseTensor

SCALAR_SHAPE

Method Summary

Modifier and Type	Method and Description
java.lang.Integer[]	asFlatArray()
double[]	asFlatDoubleArray()
int[]	asFlatIntegerArray()
long[]	asFlatLongArray()
protected JVMIntegerTensor	create(IntegerBuffer.PrimitiveIntegerWrapper buffer, long[] shape, long[] stride)
protected JVMTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>	getAsJVMTensor(IntegerTensor that)
protected JVMBuffer.PrimitiveNumberWrapperFactory<java.lang.Integer,IntegerBuffer.PrimitiveIntegerWrapper>	getFactory()
protected io.improbable.keanu.tensor.FixedPointScalarOperations<java.lang.Integer>	getOperations()
IntegerTensor	matrixMultiply(IntegerTensor that, boolean transposeLeft, boolean transposeRight)
protected IntegerTensor	set(IntegerBuffer.PrimitiveIntegerWrapper buffer, long[] shape, long[] stride)
IntegerTensor	setWithMaskInPlace(IntegerTensor mask, java.lang.Integer value)
BooleanTensor	toBoolean()
DoubleTensor	toDouble()
IntegerTensor	toInteger()
LongTensor	toLong()

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

modInPlace, modInPlace

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

absInPlace, applyInPlace, argMax, argMax, argMin, argMin, clampInPlace, cumProdInPlace, cumSumInPlace, divInPlace, divInPlace, elementwiseEquals, equalsWithinEpsilon, greaterThan, greaterThan, greaterThanMask, greaterThanOrEqual, greaterThanOrEqual, greaterThanOrEqualToMask, lessThan, lessThan, lessThanMask, lessThanOrEqual, lessThanOrEqual, lessThanOrEqualToMask, max, maxInPlace, min, minInPlace, minusInPlace, minusInPlace, plusInPlace, plusInPlace, powInPlace, powInPlace, product, product, reverseDivInPlace, reverseDivInPlace, reverseMinusInPlace, reverseMinusInPlace, setAllInPlace, signInPlace, sum, sum, sumNumber, tensorMultiply, timesInPlace, timesInPlace, unaryMinusInPlace

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

argCompare, argCompare, argCompare, argCompare, broadcast, broadcastableBinaryOpToBooleanWithAutoBroadcast, broadcastableBinaryOpWithAutoBroadcast, broadcastableBinaryOpWithAutoBroadcastInPlace, concat, concatOnDimensionZero, diag, diag, diagPart, diagPart, duplicate, elementwiseEquals, equals, fillTriangular, get, getBuffer, getFlattenedView, getLength, getRank, getShape, getStride, hashCode, isApply, permute, permute, reshape, reverseSlice, reverseSlice, slice, slice, slice, split, split, take, toString, trianglePart, triLower, triUpper, where

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

arange, arange, arange, concat, concat, create, create, create, create, create, div, eye, max, min, minus, ones, plus, pow, reverseDiv, reverseMinus, scalar, stack, times, vector, zeros

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, clamp, clampInPlace, cumProd, cumProdInPlace, cumSum, cumSumInPlace, div, div, divInPlace, divInPlace, equalsWithinEpsilon, greaterThan, greaterThan, greaterThanOrEqual, greaterThanOrEqual, lessThan, lessThan, lessThanOrEqual, lessThanOrEqual, max, maxInPlace, min, minInPlace, minus, minus, minusInPlace, minusInPlace, plus, plus, plusInPlace, plusInPlace, pow, pow, powInPlace, powInPlace, reverseDiv, reverseDiv, reverseDivInPlace, reverseDivInPlace, reverseMinus, reverseMinus, reverseMinusInPlace, reverseMinusInPlace, setAllInPlace, setWithMask, sign, signInPlace, sumNumber, times, times, timesInPlace, timesInPlace, unaryMinus, unaryMinusInPlace

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

asFlatList, create, createFilled, duplicate, elementwiseEquals, elementwiseEquals, getFlattenedView, getValue, hasShape, notEqualTo, notEqualTo, reverseSlice, scalar, setValue, split

Methods inherited from interface io.improbable.keanu.BaseNumberTensor

argMax, argMax, argMin, argMin, greaterThanMask, greaterThanOrEqualToMask, lessThanMask, lessThanOrEqualToMask, matrixMultiply, max, min, product, product, sum, sum, tensorMultiply

Methods inherited from interface io.improbable.keanu.BaseTensor

broadcast, diag, diagPart, expandDims, fillTriangular, get, getLength, getRank, getShape, getStride, isLengthOne, isMatrix, isScalar, isVector, moveAxis, permute, reshape, slice, slice, slice, sliceAlongDimension, squeeze, swapAxis, take, transpose, trianglePart, triLower, triUpper, where

Method Detail

getAsJVMTensor

protected JVMTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper> getAsJVMTensor(IntegerTensor that)

Specified by:

getAsJVMTensor in class JVMTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

getFactory

protected JVMBuffer.PrimitiveNumberWrapperFactory<java.lang.Integer,IntegerBuffer.PrimitiveIntegerWrapper> getFactory()

Specified by:

getFactory in class JVMNumberTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

getOperations

protected io.improbable.keanu.tensor.FixedPointScalarOperations<java.lang.Integer> getOperations()

Specified by:

getOperations in class JVMFixedPointTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

create

protected JVMIntegerTensor create(IntegerBuffer.PrimitiveIntegerWrapper buffer,
                                  long[] shape,
                                  long[] stride)

Specified by:

create in class JVMTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

set

protected IntegerTensor set(IntegerBuffer.PrimitiveIntegerWrapper buffer,
                            long[] shape,
                            long[] stride)

Specified by:

set in class JVMTensor<java.lang.Integer,IntegerTensor,IntegerBuffer.PrimitiveIntegerWrapper>

toBoolean

public BooleanTensor toBoolean()

Specified by:

toBoolean in interface BaseNumberTensor<BooleanTensor,IntegerTensor,DoubleTensor,java.lang.Integer,IntegerTensor>

Specified by:

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

toDouble

public DoubleTensor toDouble()

Specified by:

toDouble in interface BaseNumberTensor<BooleanTensor,IntegerTensor,DoubleTensor,java.lang.Integer,IntegerTensor>

Specified by:

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

toLong

public LongTensor toLong()

Specified by:

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

toInteger

public IntegerTensor toInteger()

Specified by:

toInteger in interface BaseNumberTensor<BooleanTensor,IntegerTensor,DoubleTensor,java.lang.Integer,IntegerTensor>

Specified by:

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

matrixMultiply

public IntegerTensor matrixMultiply(IntegerTensor that,
                                    boolean transposeLeft,
                                    boolean transposeRight)

Specified by:

matrixMultiply in interface BaseNumberTensor<BooleanTensor,IntegerTensor,DoubleTensor,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>

asFlatLongArray

public long[] asFlatLongArray()

Specified by:

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

asFlatArray

public java.lang.Integer[] asFlatArray()

Specified by:

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

setWithMaskInPlace

public IntegerTensor setWithMaskInPlace(IntegerTensor mask,
                                        java.lang.Integer value)

Specified by:

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