kmath-tensorflow/space.kscience.kmath.tensorflow/TensorFlowAlgebra

TensorFlowAlgebra

abstract class TensorFlowAlgebra<T, TT : TNumber, A : Ring<T>> : TensorAlgebra<T, A>

Functions

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

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

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>?

diagonalEmbedding

open override fun diagonalEmbedding(diagonalEntries: Tensor<T>, offset: Int, dim1: Int, dim2: Int): TensorFlowOutput<T, TT>

open infix override fun StructureND<T>.dot(other: StructureND<T>): TensorFlowOutput<T, TT>

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

open operator override fun Tensor<T>.get(i: Int): Tensor<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 fun StructureND<T>.map(transform: A.(T) -> T): StructureND<T>

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

open override fun StructureND<T>.max(): T

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

open override fun StructureND<T>.min(): T

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

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

open operator override fun StructureND<T>.minus(arg: T): TensorFlowOutput<T, TT>

open operator override fun StructureND<T>.minus(arg: StructureND<T>): TensorFlowOutput<T, TT>

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>

fun StructureND<T>.operate(operation: (Operand<TT>) -> Operand<TT>): TensorFlowOutput<T, TT>

fun T.operate(other: StructureND<T>, operation: (left: Operand<TT>, right: Operand<TT>) -> Operand<TT>): TensorFlowOutput<T, TT>

fun StructureND<T>.operate(value: T, operation: (left: Operand<TT>, right: Operand<TT>) -> Operand<TT>): TensorFlowOutput<T, TT>

fun StructureND<T>.operate(other: StructureND<T>, operation: (left: Operand<TT>, right: Operand<TT>) -> Operand<TT>): TensorFlowOutput<T, TT>

Perform binary lazy operation on tensor. Both arguments are implicitly converted

fun Tensor<T>.operateInPlace(value: T, operation: (left: Operand<TT>, right: Operand<TT>) -> Operand<TT>)

fun Tensor<T>.operateInPlace(other: StructureND<T>, operation: (left: Operand<TT>, right: Operand<TT>) -> Operand<TT>)

open operator override fun T.plus(arg: StructureND<T>): TensorFlowOutput<T, TT>

open operator override fun StructureND<T>.plus(arg: T): TensorFlowOutput<T, TT>

open operator override fun StructureND<T>.plus(arg: StructureND<T>): TensorFlowOutput<T, TT>

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

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

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

open override fun StructureND<T>.sum(): T

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

open operator override fun T.times(arg: StructureND<T>): TensorFlowOutput<T, TT>

open operator override fun StructureND<T>.times(arg: T): TensorFlowOutput<T, TT>

open operator override fun StructureND<T>.times(arg: StructureND<T>): TensorFlowOutput<T, TT>

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): Tensor<T>

open operator override fun StructureND<T>.unaryMinus(): TensorFlowOutput<T, TT>

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

open override fun StructureND<T>.valueOrNull(): T?

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

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

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

Properties

abstract val elementAlgebra: A

val ops: Ops

Inheritors

DoubleTensorFlowAlgebra

Extensions

fun <T, TT : TNumber, A : TrigonometricOperations<T>, Ring<T>> TensorFlowAlgebra<T, TT, A>.cos(arg: StructureND<T>): TensorFlowOutput<T, TT>

fun <T, TT : TNumber, A : TrigonometricOperations<T>, Ring<T>> TensorFlowAlgebra<T, TT, A>.sin(arg: StructureND<T>): TensorFlowOutput<T, TT>

Sources