kmath-tensorflow/space.kscience.kmath.tensorflow

Package space.kscience.kmath.tensorflow

Types

DoubleTensorFlowAlgebra

class DoubleTensorFlowAlgebra : TensorFlowAlgebra<Double, TFloat64, DoubleField> , PowerOperations<StructureND<Double>>

DoubleTensorFlowOutput

class DoubleTensorFlowOutput(graph: Graph, output: Output<TFloat64>) : TensorFlowOutput<Double, TFloat64>

IntTensorFlowOutput

class IntTensorFlowOutput(graph: Graph, output: Output<TInt32>) : TensorFlowOutput<Int, TInt32>

LongTensorFlowOutput

class LongTensorFlowOutput(graph: Graph, output: Output<TInt64>) : TensorFlowOutput<Long, TInt64>

TensorFlowAlgebra

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

TensorFlowArray

value class TensorFlowArray<T>(tensor: NdArray<T>) : MutableStructureND<T>

Static (eager) in-memory TensorFlow tensor

TensorFlowOutput

abstract class TensorFlowOutput<T, TT : TType>(graph: Graph, output: Output<TT>) : TensorFlowTensor<T>

Lazy graph-based TensorFlow tensor. The tensor is actualized on call.

TensorFlowTensor

interface TensorFlowTensor<T> : MutableStructureND<T>

Functions

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

produceMapWithTF

fun DoubleField.produceMapWithTF(block: DoubleTensorFlowAlgebra.() -> Map<Symbol, StructureND<Double>>): Map<Symbol, StructureND<Double>>

Compute several outputs with TensorFlow in a single run.

fun DoubleField.produceWithTF(block: DoubleTensorFlowAlgebra.() -> StructureND<Double>): StructureND<Double>

Compute a tensor with TensorFlow in a single run.

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