public class Nd4jDoubleTensor extends java.lang.Object implements DoubleTensor
Tensor.FlattenedView<T>
MINUS_ONE_SCALAR, ONE_SCALAR, TWO_SCALAR, ZERO_SCALAR
ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE
Constructor and Description |
---|
Nd4jDoubleTensor(double[] data,
long[] shape) |
Nd4jDoubleTensor(org.nd4j.linalg.api.ndarray.INDArray tensor) |
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
concat, concat, create, isNaN, max, min, reverseDiv, reverseMinus, sliceAlongDimension, stack, vector
asFlatList, create, createFilled, elementwiseEquals, getValue, hasSameShapeAs, hasSameShapeAs, isLengthOne, isMatrix, isScalar, isVector
public Nd4jDoubleTensor(double[] data, long[] shape)
public Nd4jDoubleTensor(org.nd4j.linalg.api.ndarray.INDArray tensor)
public static Nd4jDoubleTensor scalar(double scalarValue)
scalar
in interface DoubleTensor
scalar
in interface Tensor<java.lang.Double>
public static Nd4jDoubleTensor create(double[] values, long[] shape)
create
in interface DoubleTensor
public static Nd4jDoubleTensor create(double value, long[] shape)
create
in interface DoubleTensor
public static Nd4jDoubleTensor ones(long... shape)
ones
in interface DoubleTensor
public static Nd4jDoubleTensor eye(long n)
eye
in interface DoubleTensor
public static Nd4jDoubleTensor zeros(long[] shape)
zeros
in interface DoubleTensor
public static Nd4jDoubleTensor linspace(double start, double end, int numberOfPoints)
linspace
in interface DoubleTensor
public static Nd4jDoubleTensor arange(double start, double end)
arange
in interface DoubleTensor
start
- start of rangeend
- end of range (exclusive)public static Nd4jDoubleTensor arange(double start, double end, double stepSize)
arange
in interface DoubleTensor
start
- start of rangeend
- end of range (exclusive)stepSize
- size of step from start to endpublic boolean isShapePlaceholder()
isShapePlaceholder
in interface Tensor<java.lang.Double>
public java.lang.Double getValue(long... index)
public DoubleTensor setValue(java.lang.Double value, long... index)
setValue
in interface DoubleTensor
setValue
in interface Tensor<java.lang.Double>
public DoubleTensor reshape(long... newShape)
reshape
in interface DoubleTensor
reshape
in interface Tensor<java.lang.Double>
public DoubleTensor permute(int... rearrange)
permute
in interface DoubleTensor
public DoubleTensor diag()
diag
in interface DoubleTensor
diag
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor transpose()
transpose
in interface DoubleTensor
transpose
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor sum(int... overDimensions)
sum
in interface DoubleTensor
sum
in interface NumberTensor<java.lang.Double,DoubleTensor>
public java.lang.Double sum()
sum
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor duplicate()
duplicate
in interface DoubleTensor
duplicate
in interface Tensor<java.lang.Double>
public DoubleTensor apply(java.util.function.Function<java.lang.Double,java.lang.Double> function)
apply
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor matrixInverse()
matrixInverse
in interface DoubleTensor
public double max()
max
in interface DoubleTensor
public double min()
min
in interface DoubleTensor
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.Double,DoubleTensor>
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.Double,DoubleTensor>
axis
- The axis (dimension) to find the largest values inpublic double average()
average
in interface DoubleTensor
public double standardDeviation()
standardDeviation
in interface DoubleTensor
public boolean equalsWithinEpsilon(DoubleTensor o, double epsilon)
equalsWithinEpsilon
in interface DoubleTensor
public DoubleTensor clamp(DoubleTensor min, DoubleTensor max)
clamp
in interface DoubleTensor
public DoubleTensor ceil()
ceil
in interface DoubleTensor
public DoubleTensor floor()
floor
in interface DoubleTensor
public DoubleTensor round()
round
in interface DoubleTensor
public DoubleTensor standardize()
standardize
in interface DoubleTensor
public DoubleTensor replaceNaN(double value)
replaceNaN
in interface DoubleTensor
public DoubleTensor sigmoid()
sigmoid
in interface DoubleTensor
public DoubleTensor choleskyDecomposition()
choleskyDecomposition
in interface DoubleTensor
public DoubleTensor reciprocal()
reciprocal
in interface DoubleTensor
public DoubleTensor minus(double value)
minus
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
minus
in interface DoubleTensor
public DoubleTensor plus(double value)
plus
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
plus
in interface DoubleTensor
public DoubleTensor times(double value)
times
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
times
in interface DoubleTensor
public DoubleTensor matrixMultiply(DoubleTensor value)
matrixMultiply
in interface DoubleTensor
matrixMultiply
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor tensorMultiply(DoubleTensor value, int[] dimsLeft, int[] dimsRight)
tensorMultiply
in interface DoubleTensor
tensorMultiply
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor div(double value)
div
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
div
in interface DoubleTensor
public DoubleTensor pow(DoubleTensor exponent)
pow
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
pow
in interface DoubleTensor
public DoubleTensor pow(double exponent)
pow
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
pow
in interface DoubleTensor
public DoubleTensor sqrt()
sqrt
in interface DoubleTensor
public DoubleTensor log()
log
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
log
in interface DoubleTensor
public DoubleTensor safeLogTimes(DoubleTensor y)
safeLogTimes
in interface DoubleTensor
public DoubleTensor logGamma()
logGamma
in interface DoubleTensor
public DoubleTensor digamma()
digamma
in interface DoubleTensor
public DoubleTensor sin()
sin
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
sin
in interface DoubleTensor
public DoubleTensor cos()
cos
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
cos
in interface DoubleTensor
public DoubleTensor tan()
tan
in interface DoubleTensor
public DoubleTensor atan()
atan
in interface DoubleTensor
public DoubleTensor atan2(double y)
atan2
in interface DoubleTensor
public DoubleTensor atan2(DoubleTensor y)
atan2
in interface DoubleTensor
public DoubleTensor asin()
asin
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
asin
in interface DoubleTensor
public DoubleTensor acos()
acos
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
acos
in interface DoubleTensor
public DoubleTensor exp()
exp
in interface io.improbable.keanu.kotlin.DoubleOperators<DoubleTensor>
exp
in interface DoubleTensor
public DoubleTensor minus(DoubleTensor that)
minus
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
public DoubleTensor plus(DoubleTensor that)
plus
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
public DoubleTensor times(DoubleTensor that)
times
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
public DoubleTensor div(DoubleTensor that)
div
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
public DoubleTensor abs()
abs
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor unaryMinus()
unaryMinus
in interface io.improbable.keanu.kotlin.NumberOperators<DoubleTensor>
public DoubleTensor setWithMask(DoubleTensor mask, java.lang.Double value)
setWithMask
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor reciprocalInPlace()
reciprocalInPlace
in interface DoubleTensor
public DoubleTensor minusInPlace(double value)
minusInPlace
in interface DoubleTensor
public DoubleTensor plusInPlace(double value)
plusInPlace
in interface DoubleTensor
public DoubleTensor timesInPlace(double value)
timesInPlace
in interface DoubleTensor
public DoubleTensor divInPlace(double value)
divInPlace
in interface DoubleTensor
public DoubleTensor powInPlace(DoubleTensor exponent)
powInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor powInPlace(double exponent)
powInPlace
in interface DoubleTensor
public DoubleTensor sqrtInPlace()
sqrtInPlace
in interface DoubleTensor
public DoubleTensor logInPlace()
logInPlace
in interface DoubleTensor
public DoubleTensor safeLogTimesInPlace(DoubleTensor y)
safeLogTimesInPlace
in interface DoubleTensor
y
- The tensor value to multiply bypublic DoubleTensor logGammaInPlace()
logGammaInPlace
in interface DoubleTensor
public DoubleTensor digammaInPlace()
digammaInPlace
in interface DoubleTensor
public DoubleTensor sinInPlace()
sinInPlace
in interface DoubleTensor
public DoubleTensor cosInPlace()
cosInPlace
in interface DoubleTensor
public DoubleTensor tanInPlace()
tanInPlace
in interface DoubleTensor
public DoubleTensor atanInPlace()
atanInPlace
in interface DoubleTensor
public DoubleTensor atan2InPlace(double y)
atan2InPlace
in interface DoubleTensor
public DoubleTensor atan2InPlace(DoubleTensor y)
atan2InPlace
in interface DoubleTensor
public DoubleTensor asinInPlace()
asinInPlace
in interface DoubleTensor
public DoubleTensor acosInPlace()
acosInPlace
in interface DoubleTensor
public DoubleTensor expInPlace()
expInPlace
in interface DoubleTensor
public DoubleTensor minusInPlace(DoubleTensor that)
minusInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
that
- Right operand.public DoubleTensor plusInPlace(DoubleTensor that)
plusInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
that
- Right operand.public DoubleTensor timesInPlace(DoubleTensor that)
timesInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
that
- Right operand.public DoubleTensor divInPlace(DoubleTensor that)
divInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
that
- Right operand.public DoubleTensor unaryMinusInPlace()
unaryMinusInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor absInPlace()
absInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor getGreaterThanMask(DoubleTensor greaterThanThis)
getGreaterThanMask
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor getGreaterThanOrEqualToMask(DoubleTensor greaterThanOrEqualToThis)
getGreaterThanOrEqualToMask
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor getLessThanMask(DoubleTensor lessThanThis)
getLessThanMask
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor getLessThanOrEqualToMask(DoubleTensor lessThanOrEqualToThis)
getLessThanOrEqualToMask
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor setWithMaskInPlace(DoubleTensor mask, java.lang.Double value)
setWithMaskInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor applyInPlace(java.util.function.Function<java.lang.Double,java.lang.Double> function)
applyInPlace
in interface NumberTensor<java.lang.Double,DoubleTensor>
public DoubleTensor maxInPlace(DoubleTensor max)
maxInPlace
in interface DoubleTensor
public DoubleTensor minInPlace(DoubleTensor min)
minInPlace
in interface DoubleTensor
public DoubleTensor standardizeInPlace()
standardizeInPlace
in interface DoubleTensor
public DoubleTensor setAllInPlace(double value)
setAllInPlace
in interface DoubleTensor
public DoubleTensor clampInPlace(DoubleTensor min, DoubleTensor max)
clampInPlace
in interface DoubleTensor
public DoubleTensor ceilInPlace()
ceilInPlace
in interface DoubleTensor
public DoubleTensor floorInPlace()
floorInPlace
in interface DoubleTensor
public DoubleTensor roundInPlace()
roundInPlace
in interface DoubleTensor
public DoubleTensor sigmoidInPlace()
sigmoidInPlace
in interface DoubleTensor
public double determinant()
determinant
in interface DoubleTensor
public double product()
product
in interface DoubleTensor
public DoubleTensor slice(int dimension, long index)
slice
in interface DoubleTensor
slice
in interface Tensor<java.lang.Double>
public java.util.List<DoubleTensor> split(int dimension, long... splitAtIndices)
split
in interface DoubleTensor
dimension
- the dimension to slice onsplitAtIndices
- the indices that the dimension to slice on should be slice one.g A = [ 1, 2, 3, 4, 5, 6 7, 8, 9, 1, 2, 3 ]
A.slice(0, [1]) gives List([1, 2, 3, 4, 5, 6]) A.slice(0, [1, 2]) gives List([1, 2, 3, 4, 5, 6], [7, 8, 9, 1, 2, 3]
A.slice(1, [1, 3, 6]) gives List( [1, [2, 3 , [4, 5, 6, 7] 8, 9] 1, 2, 3] )
public BooleanTensor lessThan(double value)
lessThan
in interface DoubleTensor
public BooleanTensor lessThanOrEqual(double value)
lessThanOrEqual
in interface DoubleTensor
public BooleanTensor lessThan(DoubleTensor value)
lessThan
in interface NumberTensor<java.lang.Double,DoubleTensor>
public BooleanTensor lessThanOrEqual(DoubleTensor value)
lessThanOrEqual
in interface NumberTensor<java.lang.Double,DoubleTensor>
public BooleanTensor greaterThan(double value)
greaterThan
in interface DoubleTensor
public BooleanTensor greaterThanOrEqual(double value)
greaterThanOrEqual
in interface DoubleTensor
public BooleanTensor notNaN()
notNaN
in interface DoubleTensor
public DoubleTensor replaceNaNInPlace(double value)
replaceNaNInPlace
in interface DoubleTensor
public BooleanTensor greaterThan(DoubleTensor value)
greaterThan
in interface NumberTensor<java.lang.Double,DoubleTensor>
public BooleanTensor greaterThanOrEqual(DoubleTensor value)
greaterThanOrEqual
in interface NumberTensor<java.lang.Double,DoubleTensor>
public BooleanTensor elementwiseEquals(Tensor that)
elementwiseEquals
in interface Tensor<java.lang.Double>
public BooleanTensor elementwiseEquals(java.lang.Double value)
elementwiseEquals
in interface Tensor<java.lang.Double>
public Tensor.FlattenedView<java.lang.Double> getFlattenedView()
getFlattenedView
in interface Tensor<java.lang.Double>
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 DoubleTensor toDouble()
toDouble
in interface NumberTensor<java.lang.Double,DoubleTensor>
public IntegerTensor toInteger()
toInteger
in interface NumberTensor<java.lang.Double,DoubleTensor>
public double[] asFlatDoubleArray()
asFlatDoubleArray
in interface Tensor<java.lang.Double>
public int[] asFlatIntegerArray()
asFlatIntegerArray
in interface Tensor<java.lang.Double>
public java.lang.Double[] asFlatArray()
asFlatArray
in interface Tensor<java.lang.Double>