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

Nd4jTensorAlgebra

interface Nd4jTensorAlgebra<T : Number, A : Field<T>> : AnalyticTensorAlgebra<T, A> (source)

ND4J based TensorAlgebra implementation.

Inheritors

DoubleNd4jTensorAlgebra

Functions

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

open override fun acosh(arg: StructureND<T>): Nd4jArrayStructure<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>.argMin(dim: Int, keepDim: Boolean): Tensor<Int>

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

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

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

open override fun atanh(arg: StructureND<T>): Nd4jArrayStructure<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 ceil(arg: StructureND<T>): Nd4jArrayStructure<T>

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

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

diagonalEmbedding

abstract fun diagonalEmbedding(diagonalEntries: StructureND<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 exp(arg: StructureND<T>): Nd4jArrayStructure<T>

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

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

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

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

open fun Tensor<T>.getTensor(first: Int, second: Int): Tensor<T>

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

open override fun ln(arg: StructureND<T>): Nd4jArrayStructure<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>

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

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

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

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

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

abstract fun StructureND<T>.min(): 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 infix fun StructureND<T>.pow(pow: Number): StructureND<T>

open override fun power(arg: StructureND<T>, pow: Number): StructureND<T>

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

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

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

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

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

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

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

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

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

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

open override fun tanh(arg: StructureND<T>): Nd4jArrayStructure<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 StructureND<T>.transposed(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?

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

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

open override fun Tensor<T>.view(shape: ShapeND): 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

open val bufferFactory: MutableBufferFactory<StructureND<T>>

abstract val elementAlgebra: A

abstract val StructureND<T>.ndArray: INDArray

Unwraps to or gets INDArray from StructureND.