public interface NumberTensor<N extends java.lang.Number,T extends NumberTensor<N,T>> extends Tensor<N,T>, io.improbable.keanu.kotlin.NumberOperators<T>
Tensor.FlattenedView<N>ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE| Modifier and Type | Method and Description |
|---|---|
default T |
abs() |
T |
absInPlace() |
default T |
apply(java.util.function.Function<N,N> function) |
T |
applyInPlace(java.util.function.Function<N,N> function) |
int |
argMax()
Find the index into the flattened array of the tensor of the largest value, e.g.
|
IntegerTensor |
argMax(int axis)
Find the indices into the tensor of the largest values in a specified axis (dimension), e.g.
|
int |
argMin() |
IntegerTensor |
argMin(int axis) |
double[] |
asFlatDoubleArray() |
int[] |
asFlatIntegerArray() |
N |
average() |
default T |
clamp(T min,
T max) |
T |
clampInPlace(T min,
T max) |
default T |
cumProd(int requestedDimension) |
T |
cumProdInPlace(int dimension) |
default T |
cumSum(int requestedDimension) |
T |
cumSumInPlace(int dimension) |
default T |
div(N value) |
default T |
div(T value) |
T |
divInPlace(N value) |
T |
divInPlace(T that) |
boolean |
equalsWithinEpsilon(T other,
N epsilon) |
BooleanTensor |
greaterThan(N value) |
BooleanTensor |
greaterThan(T value) |
T |
greaterThanMask(T greaterThanThis) |
BooleanTensor |
greaterThanOrEqual(N value) |
BooleanTensor |
greaterThanOrEqual(T value) |
T |
greaterThanOrEqualToMask(T greaterThanThis) |
BooleanTensor |
lessThan(N value) |
BooleanTensor |
lessThan(T value) |
T |
lessThanMask(T lessThanThis) |
BooleanTensor |
lessThanOrEqual(N value) |
BooleanTensor |
lessThanOrEqual(T value) |
T |
lessThanOrEqualToMask(T lessThanThis) |
T |
matrixMultiply(T value) |
N |
max() |
default T |
max(T max) |
T |
maxInPlace(T max) |
N |
min() |
default T |
min(T min) |
T |
minInPlace(T min) |
default T |
minus(N value) |
default T |
minus(T that) |
T |
minusInPlace(N value) |
T |
minusInPlace(T that) |
default T |
plus(N value) |
default T |
plus(T that) |
T |
plusInPlace(N value) |
T |
plusInPlace(T that) |
default T |
pow(N exponent) |
default T |
pow(T exponent) |
T |
powInPlace(N exponent) |
T |
powInPlace(T exponent) |
N |
product() |
T |
product(int... overDimensions) |
default T |
reverseDiv(N value) |
default T |
reverseDiv(T value) |
T |
reverseDivInPlace(N value) |
T |
reverseDivInPlace(T value) |
default T |
reverseMinus(N value) |
default T |
reverseMinus(T value) |
T |
reverseMinusInPlace(N value) |
T |
reverseMinusInPlace(T value) |
default T |
safeLogTimes(T y)
This is identical to log().times(y), except that it changes NaN results to 0.
|
T |
safeLogTimesInPlace(T y) |
T |
setAllInPlace(N value) |
default T |
setWithMask(T mask,
N value) |
T |
setWithMaskInPlace(T mask,
N value) |
N |
standardDeviation() |
N |
sum() |
T |
sum(int... overDimensions) |
T |
tensorMultiply(T value,
int[] dimLeft,
int[] dimsRight) |
default T |
times(N value) |
default T |
times(T that) |
T |
timesInPlace(N value) |
T |
timesInPlace(T that) |
DoubleTensor |
toDouble() |
IntegerTensor |
toInteger() |
default T |
unaryMinus() |
T |
unaryMinusInPlace() |
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, transposeDoubleTensor toDouble()
IntegerTensor toInteger()
double[] asFlatDoubleArray()
int[] asFlatIntegerArray()
N sum()
T sum(int... overDimensions)
default T cumSum(int requestedDimension)
T cumSumInPlace(int dimension)
N product()
T product(int... overDimensions)
default T cumProd(int requestedDimension)
T cumProdInPlace(int dimension)
N max()
N min()
default T abs()
T absInPlace()
default T minus(T that)
minus in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>default T plus(T that)
plus in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>default T unaryMinus()
unaryMinus in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>T unaryMinusInPlace()
default T times(T that)
times in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>default T div(T value)
div in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>default T pow(T exponent)
pow in interface io.improbable.keanu.kotlin.NumberOperators<T extends NumberTensor<N,T>>N average()
N standardDeviation()
int argMax()
DoubleTensor tensor = DoubleTensor.arange(0, 6).reshape(2, 3); // [[0., 1., 2.], // [3., 4., 5.]] IntegerTensor max = tensor.argMax(); // [[5]]
IntegerTensor argMax(int axis)
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]]
axis - The axis (dimension) to find the largest values inint argMin()
IntegerTensor argMin(int axis)
default T safeLogTimes(T y)
y - The tensor value to multiply byBooleanTensor lessThan(T value)
BooleanTensor lessThanOrEqual(T value)
BooleanTensor greaterThan(T value)
BooleanTensor greaterThanOrEqual(T value)
BooleanTensor lessThan(N value)
BooleanTensor lessThanOrEqual(N value)
BooleanTensor greaterThan(N value)
BooleanTensor greaterThanOrEqual(N value)