public class Nd4jIntegerTensor extends java.lang.Object implements IntegerTensor
Tensor.FlattenedView<T>
ONE_SCALAR, TWO_SCALAR, ZERO_SCALAR
ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE
Constructor and Description |
---|
Nd4jIntegerTensor(org.nd4j.linalg.api.ndarray.INDArray tensor) |
Nd4jIntegerTensor(int[] data,
long[] shape) |
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
concat, concat, create, eye, max, min, reverseDiv, reverseMinus, stack
asFlatList, create, createFilled, elementwiseEquals, getValue, hasSameShapeAs, hasSameShapeAs, isLengthOne, isMatrix, isScalar, isVector
public Nd4jIntegerTensor(int[] data, long[] 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, long[] shape)
create
in interface IntegerTensor
public static Nd4jIntegerTensor create(int value, long[] shape)
create
in interface IntegerTensor
public static Nd4jIntegerTensor ones(long[] shape)
ones
in interface IntegerTensor
public static Nd4jIntegerTensor eye(long n)
public static Nd4jIntegerTensor zeros(long[] shape)
zeros
in interface IntegerTensor
public IntegerTensor reshape(long... newShape)
reshape
in interface IntegerTensor
reshape
in interface Tensor<java.lang.Integer>
public IntegerTensor diag()
diag
in interface IntegerTensor
diag
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor transpose()
transpose
in interface IntegerTensor
transpose
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor sum(int... overDimensions)
sum
in interface IntegerTensor
sum
in interface NumberTensor<java.lang.Integer,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
matrixMultiply
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor tensorMultiply(IntegerTensor value, int[] dimLeft, int[] dimsRight)
tensorMultiply
in interface IntegerTensor
tensorMultiply
in interface NumberTensor<java.lang.Integer,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 io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
public IntegerTensor pow(int exponent)
pow
in interface io.improbable.keanu.kotlin.IntegerOperators<IntegerTensor>
pow
in interface IntegerTensor
public IntegerTensor minus(IntegerTensor that)
minus
in interface io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
minus
in interface IntegerTensor
public IntegerTensor plus(IntegerTensor that)
plus
in interface io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
plus
in interface IntegerTensor
public IntegerTensor times(IntegerTensor that)
times
in interface io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
times
in interface IntegerTensor
public IntegerTensor div(IntegerTensor that)
div
in interface io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
div
in interface IntegerTensor
public IntegerTensor unaryMinus()
unaryMinus
in interface io.improbable.keanu.kotlin.NumberOperators<IntegerTensor>
unaryMinus
in interface IntegerTensor
public IntegerTensor abs()
abs
in interface IntegerTensor
abs
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor setWithMask(IntegerTensor mask, java.lang.Integer value)
setWithMask
in interface IntegerTensor
setWithMask
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor getGreaterThanMask(IntegerTensor greaterThanThis)
getGreaterThanMask
in interface IntegerTensor
getGreaterThanMask
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor getGreaterThanOrEqualToMask(IntegerTensor greaterThanOrEqualToThis)
getGreaterThanOrEqualToMask
in interface IntegerTensor
getGreaterThanOrEqualToMask
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor getLessThanMask(IntegerTensor lessThanThis)
getLessThanMask
in interface IntegerTensor
getLessThanMask
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor getLessThanOrEqualToMask(IntegerTensor lessThanOrEqualToThis)
getLessThanOrEqualToMask
in interface IntegerTensor
getLessThanOrEqualToMask
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor setWithMaskInPlace(IntegerTensor mask, java.lang.Integer value)
setWithMaskInPlace
in interface IntegerTensor
setWithMaskInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor apply(java.util.function.Function<java.lang.Integer,java.lang.Integer> function)
apply
in interface IntegerTensor
apply
in interface NumberTensor<java.lang.Integer,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 NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor powInPlace(int exponent)
powInPlace
in interface IntegerTensor
public IntegerTensor minusInPlace(IntegerTensor that)
minusInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor plusInPlace(IntegerTensor that)
plusInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor timesInPlace(IntegerTensor that)
timesInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor divInPlace(IntegerTensor that)
divInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor unaryMinusInPlace()
unaryMinusInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor absInPlace()
absInPlace
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor applyInPlace(java.util.function.Function<java.lang.Integer,java.lang.Integer> function)
applyInPlace
in interface NumberTensor<java.lang.Integer,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 NumberTensor<java.lang.Integer,IntegerTensor>
public BooleanTensor lessThanOrEqual(IntegerTensor value)
lessThanOrEqual
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public BooleanTensor greaterThan(int value)
greaterThan
in interface IntegerTensor
public BooleanTensor greaterThanOrEqual(int value)
greaterThanOrEqual
in interface IntegerTensor
public IntegerTensor minInPlace(IntegerTensor min)
minInPlace
in interface IntegerTensor
public IntegerTensor maxInPlace(IntegerTensor max)
maxInPlace
in interface IntegerTensor
public int min()
min
in interface IntegerTensor
public int max()
max
in interface IntegerTensor
public int argMax()
NumberTensor
DoubleTensor tensor = DoubleTensor.arange(0, 6).reshape(2, 3); // [[0., 1., 2.], // [3., 4., 5.]] IntegerTensor max = tensor.argMax(); // [[5]]
argMax
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor argMax(int axis)
NumberTensor
DoubleTensor tensor = DoubleTensor.arange(0, 6).reshape(2, 3); // [[0., 1., 2.], // [3., 4., 5.]] IntegerTensor maxesFor0 = tensor.argMax(0); // [[1, 1, 1]] IntegerTensor maxFor1 = tensor.argMax(1); // [[2, 2]]
argMax
in interface NumberTensor<java.lang.Integer,IntegerTensor>
axis
- The axis (dimension) to find the largest values inpublic BooleanTensor greaterThan(IntegerTensor value)
greaterThan
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public BooleanTensor greaterThanOrEqual(IntegerTensor value)
greaterThanOrEqual
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public java.lang.Integer sum()
sum
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public DoubleTensor toDouble()
toDouble
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public IntegerTensor toInteger()
toInteger
in interface NumberTensor<java.lang.Integer,IntegerTensor>
public boolean isShapePlaceholder()
isShapePlaceholder
in interface Tensor<java.lang.Integer>
public java.lang.Integer getValue(long... index)
public IntegerTensor setValue(java.lang.Integer value, long... index)
setValue
in interface IntegerTensor
setValue
in interface Tensor<java.lang.Integer>
public java.lang.Integer scalar()
public IntegerTensor duplicate()
duplicate
in interface IntegerTensor
duplicate
in interface Tensor<java.lang.Integer>
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 BooleanTensor elementwiseEquals(java.lang.Integer value)
elementwiseEquals
in interface Tensor<java.lang.Integer>
public IntegerTensor slice(int dimension, long index)
slice
in interface IntegerTensor
slice
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 java.lang.String toString()
toString
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>