kmath-nd4j/space.kscience.kmath.nd4j/Nd4jTensorAlgebra

Nd4jTensorAlgebra

interface Nd4jTensorAlgebra<T : Number, A : Field<T>> : AnalyticTensorAlgebra<T, A>

ND4J based TensorAlgebra implementation.

Functions

open override fun StructureND<T>.acos(): Nd4jArrayStructure<T>

open override fun acos(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.acosh(): Nd4jArrayStructure<T>

open override fun acosh(arg: StructureND<T>): StructureND<T>

open override fun add(left: StructureND<T>, right: StructureND<T>): Tensor<T>

open override fun StructureND<T>.argMax(dim: Int, keepDim: Boolean): Tensor<Int>

open override fun StructureND<T>.asin(): Nd4jArrayStructure<T>

open override fun asin(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.asinh(): Nd4jArrayStructure<T>

open override fun asinh(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.atan(): Nd4jArrayStructure<T>

open override fun atan(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.atanh(): Nd4jArrayStructure<T>

open override fun atanh(arg: StructureND<T>): StructureND<T>

binaryOperation

open fun binaryOperation(operation: String, left: StructureND<T>, right: StructureND<T>): StructureND<T>

binaryOperationFunction

open override fun binaryOperationFunction(operation: String): (left: StructureND<T>, right: StructureND<T>) -> StructureND<T>

open fun bindSymbol(value: String): StructureND<T>

bindSymbolOrNull

open fun bindSymbolOrNull(value: String): StructureND<T>?

open override fun StructureND<T>.ceil(): Nd4jArrayStructure<T>

open override fun StructureND<T>.cos(): Nd4jArrayStructure<T>

open override fun cos(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.cosh(): Nd4jArrayStructure<T>

open override fun cosh(arg: StructureND<T>): StructureND<T>

diagonalEmbedding

abstract fun diagonalEmbedding(diagonalEntries: Tensor<T>, offset: Int, dim1: Int, dim2: Int): Tensor<T>

open operator override fun T.div(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.div(arg: T): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.div(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator fun StructureND<T>.div(k: Number): StructureND<T>

open operator override fun Tensor<T>.divAssign(value: T)

open operator override fun Tensor<T>.divAssign(arg: StructureND<T>)

open override fun divide(left: StructureND<T>, right: StructureND<T>): StructureND<T>

open infix override fun StructureND<T>.dot(other: StructureND<T>): Nd4jArrayStructure<T>

open override fun StructureND<T>.exp(): Nd4jArrayStructure<T>

open override fun exp(arg: StructureND<T>): StructureND<T>

open fun export(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.floor(): Nd4jArrayStructure<T>

open operator override fun Tensor<T>.get(i: Int): Nd4jArrayStructure<T>

open fun <F : StructureFeature> getFeature(structure: StructureND<T>, type: KClass<out F>): F?

open operator fun (T) -> T.invoke(structure: StructureND<T>): StructureND<T>

open override fun StructureND<T>.ln(): Nd4jArrayStructure<T>

open override fun ln(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.map(transform: A.(T) -> T): Nd4jArrayStructure<T>

open override fun StructureND<T>.mapIndexed(transform: A.(index: IntArray, T) -> T): Nd4jArrayStructure<T>

abstract fun StructureND<T>.max(): T?

open override fun StructureND<T>.max(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

abstract fun StructureND<T>.mean(): T

open override fun StructureND<T>.mean(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

abstract fun StructureND<T>.min(): T?

open override fun StructureND<T>.min(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

open operator override fun T.minus(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.minus(arg: T): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.minus(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun Tensor<T>.minusAssign(value: T)

open operator override fun Tensor<T>.minusAssign(arg: StructureND<T>)

open override fun multiply(left: StructureND<T>, right: StructureND<T>): Tensor<T>

open operator override fun T.plus(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.plus(arg: T): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.plus(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun Tensor<T>.plusAssign(value: T)

open operator override fun Tensor<T>.plusAssign(arg: StructureND<T>)

open override fun scale(a: StructureND<T>, value: Double): StructureND<T>

open override fun StructureND<T>.sin(): Nd4jArrayStructure<T>

open override fun sin(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.sinh(): Tensor<T>

open override fun sinh(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.sqrt(): Nd4jArrayStructure<T>

abstract fun StructureND<T>.std(): T

open override fun StructureND<T>.std(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

abstract override fun structureND(shape: Shape, initializer: A.(IntArray) -> T): Nd4jArrayStructure<T>

abstract fun StructureND<T>.sum(): T

open override fun StructureND<T>.sum(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

open override fun StructureND<T>.tan(): Nd4jArrayStructure<T>

open override fun tan(arg: StructureND<T>): StructureND<T>

open override fun StructureND<T>.tanh(): Nd4jArrayStructure<T>

open override fun tanh(arg: StructureND<T>): StructureND<T>

open operator override fun T.times(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.times(arg: T): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.times(arg: StructureND<T>): Nd4jArrayStructure<T>

open operator fun Number.times(arg: StructureND<T>): StructureND<T>

open operator fun StructureND<T>.times(k: Number): StructureND<T>

open operator override fun Tensor<T>.timesAssign(value: T)

open operator override fun Tensor<T>.timesAssign(arg: StructureND<T>)

open override fun Tensor<T>.transpose(i: Int, j: Int): Nd4jArrayStructure<T>

open operator override fun StructureND<T>.unaryMinus(): Nd4jArrayStructure<T>

open fun unaryOperation(operation: String, arg: StructureND<T>): StructureND<T>

unaryOperationFunction

open override fun unaryOperationFunction(operation: String): (arg: StructureND<T>) -> StructureND<T>

open operator fun StructureND<T>.unaryPlus(): StructureND<T>

open fun StructureND<T>.value(): T

abstract fun StructureND<T>.valueOrNull(): T?

abstract fun StructureND<T>.variance(): T

open override fun StructureND<T>.variance(dim: Int, keepDim: Boolean): Nd4jArrayStructure<T>

open override fun Tensor<T>.view(shape: IntArray): Nd4jArrayStructure<T>

open override fun Tensor<T>.viewAs(other: StructureND<T>): Nd4jArrayStructure<T>

abstract fun INDArray.wrap(): Nd4jArrayStructure<T>

Wraps INDArray to Nd4jArrayStructure.

open override fun zip(left: StructureND<T>, right: StructureND<T>, transform: A.(T, T) -> T): Nd4jArrayStructure<T>

Properties

abstract val elementAlgebra: A

abstract val StructureND<T>.ndArray: INDArray

Unwraps to or gets INDArray from StructureND.

Inheritors

DoubleNd4jTensorAlgebra

Sources