public class Nd4jIntegerTensor extends java.lang.Object implements IntegerTensor
Tensor.FlattenedView<T>
ONE_SCALAR, TWO_SCALAR, ZERO_SCALAR
SCALAR_SHAPE, SCALAR_STRIDE
Constructor and Description |
---|
Nd4jIntegerTensor(org.nd4j.linalg.api.ndarray.INDArray tensor) |
Nd4jIntegerTensor(int[] data,
int[] shape) |
clone, finalize, getClass, notify, notifyAll, toString, wait, wait, wait
create, create, placeHolder
asFlatList, elementwiseEquals, hasSameShapeAs, hasSameShapeAs, isMatrix, isScalar, isVector
public Nd4jIntegerTensor(int[] data, int[] shape)
public Nd4jIntegerTensor(org.nd4j.linalg.api.ndarray.INDArray tensor)
public static Nd4jIntegerTensor scalar(int scalarValue)
scalar
in interface IntegerTensor
scalar
in interface Tensor<java.lang.Integer>
public static Nd4jIntegerTensor create(int[] values, int[] shape)
create
in interface IntegerTensor
public static Nd4jIntegerTensor create(double value, int[] shape)
public static Nd4jIntegerTensor ones(int[] shape)
ones
in interface IntegerTensor
public static Nd4jIntegerTensor eye(int n)
eye
in interface IntegerTensor
public static Nd4jIntegerTensor zeros(int[] shape)
zeros
in interface IntegerTensor
public IntegerTensor reshape(int... newShape)
reshape
in interface IntegerTensor
reshape
in interface Tensor<java.lang.Integer>
public IntegerTensor diag()
diag
in interface IntegerTensor
public IntegerTensor transpose()
transpose
in interface IntegerTensor
public IntegerTensor sum(int... overDimensions)
sum
in interface IntegerTensor
public IntegerTensor minus(int value)
minus
in interface io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>
minus
in interface IntegerTensor
public IntegerTensor plus(int value)
plus
in interface io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>
plus
in interface IntegerTensor
public IntegerTensor times(int value)
times
in interface io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>
times
in interface IntegerTensor
public IntegerTensor matrixMultiply(IntegerTensor value)
matrixMultiply
in interface IntegerTensor
public IntegerTensor tensorMultiply(IntegerTensor value, int[] dimLeft, int[] dimsRight)
tensorMultiply
in interface IntegerTensor
public IntegerTensor div(int value)
div
in interface io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>
div
in interface IntegerTensor
public IntegerTensor pow(IntegerTensor exponent)
pow
in interface IntegerTensor
public IntegerTensor pow(int exponent)
pow
in interface IntegerTensor
public IntegerTensor minus(IntegerTensor that)
minus
in interface io.improbable.keanu.kotlin.Operators<IntegerTensor>
minus
in interface IntegerTensor
public IntegerTensor plus(IntegerTensor that)
plus
in interface io.improbable.keanu.kotlin.Operators<IntegerTensor>
plus
in interface IntegerTensor
public IntegerTensor times(IntegerTensor that)
times
in interface io.improbable.keanu.kotlin.Operators<IntegerTensor>
times
in interface IntegerTensor
public IntegerTensor div(IntegerTensor that)
div
in interface io.improbable.keanu.kotlin.Operators<IntegerTensor>
div
in interface IntegerTensor
public IntegerTensor unaryMinus()
unaryMinus
in interface io.improbable.keanu.kotlin.Operators<IntegerTensor>
unaryMinus
in interface IntegerTensor
public IntegerTensor abs()
abs
in interface IntegerTensor
public IntegerTensor setWithMask(IntegerTensor mask, int value)
setWithMask
in interface IntegerTensor
public IntegerTensor getGreaterThanMask(IntegerTensor greaterThanThis)
getGreaterThanMask
in interface IntegerTensor
public IntegerTensor getGreaterThanOrEqualToMask(IntegerTensor greaterThanOrEqualToThis)
getGreaterThanOrEqualToMask
in interface IntegerTensor
public IntegerTensor getLessThanMask(IntegerTensor lessThanThis)
getLessThanMask
in interface IntegerTensor
public IntegerTensor getLessThanOrEqualToMask(IntegerTensor lessThanOrEqualToThis)
getLessThanOrEqualToMask
in interface IntegerTensor
public IntegerTensor setWithMaskInPlace(IntegerTensor mask, int value)
setWithMaskInPlace
in interface IntegerTensor
public IntegerTensor apply(java.util.function.Function<java.lang.Integer,java.lang.Integer> function)
apply
in interface IntegerTensor
public IntegerTensor minusInPlace(int value)
minusInPlace
in interface IntegerTensor
public IntegerTensor plusInPlace(int value)
plusInPlace
in interface IntegerTensor
public IntegerTensor timesInPlace(int value)
timesInPlace
in interface IntegerTensor
public IntegerTensor divInPlace(int value)
divInPlace
in interface IntegerTensor
public IntegerTensor powInPlace(IntegerTensor exponent)
powInPlace
in interface IntegerTensor
public IntegerTensor powInPlace(int exponent)
powInPlace
in interface IntegerTensor
public IntegerTensor minusInPlace(IntegerTensor that)
minusInPlace
in interface IntegerTensor
public IntegerTensor plusInPlace(IntegerTensor that)
plusInPlace
in interface IntegerTensor
public IntegerTensor timesInPlace(IntegerTensor that)
timesInPlace
in interface IntegerTensor
public IntegerTensor divInPlace(IntegerTensor that)
divInPlace
in interface IntegerTensor
public IntegerTensor unaryMinusInPlace()
unaryMinusInPlace
in interface IntegerTensor
public IntegerTensor absInPlace()
absInPlace
in interface IntegerTensor
public IntegerTensor applyInPlace(java.util.function.Function<java.lang.Integer,java.lang.Integer> function)
applyInPlace
in interface IntegerTensor
public BooleanTensor lessThan(int value)
lessThan
in interface IntegerTensor
public BooleanTensor lessThanOrEqual(int value)
lessThanOrEqual
in interface IntegerTensor
public BooleanTensor lessThan(IntegerTensor value)
lessThan
in interface IntegerTensor
public BooleanTensor lessThanOrEqual(IntegerTensor value)
lessThanOrEqual
in interface IntegerTensor
public BooleanTensor greaterThan(int value)
greaterThan
in interface IntegerTensor
public BooleanTensor greaterThanOrEqual(int value)
greaterThanOrEqual
in interface IntegerTensor
public BooleanTensor greaterThan(IntegerTensor value)
greaterThan
in interface IntegerTensor
public BooleanTensor greaterThanOrEqual(IntegerTensor value)
greaterThanOrEqual
in interface IntegerTensor
public java.lang.Integer sum()
sum
in interface NumberTensor<java.lang.Integer>
public DoubleTensor toDouble()
toDouble
in interface NumberTensor<java.lang.Integer>
public IntegerTensor toInteger()
toInteger
in interface NumberTensor<java.lang.Integer>
public boolean isShapePlaceholder()
isShapePlaceholder
in interface Tensor<java.lang.Integer>
public java.lang.Integer getValue(int... index)
public void setValue(java.lang.Integer value, int... index)
public java.lang.Integer scalar()
public IntegerTensor duplicate()
public Tensor.FlattenedView<java.lang.Integer> getFlattenedView()
getFlattenedView
in interface Tensor<java.lang.Integer>
public BooleanTensor elementwiseEquals(Tensor that)
elementwiseEquals
in interface Tensor<java.lang.Integer>
public boolean equals(java.lang.Object o)
equals
in class java.lang.Object
public int hashCode()
hashCode
in class java.lang.Object
public double[] asFlatDoubleArray()
asFlatDoubleArray
in interface Tensor<java.lang.Integer>
public int[] asFlatIntegerArray()
asFlatIntegerArray
in interface Tensor<java.lang.Integer>
public java.lang.Integer[] asFlatArray()
asFlatArray
in interface Tensor<java.lang.Integer>