Package space.kscience.kmath.tensorflow

Types

Link copied to clipboard
Link copied to clipboard
class DoubleTensorFlowOutput(graph: Graph, output: Output<TFloat64>) : TensorFlowOutput<Double, TFloat64>
Link copied to clipboard
class IntTensorFlowOutput(graph: Graph, output: Output<TInt32>) : TensorFlowOutput<Int, TInt32>
Link copied to clipboard
class LongTensorFlowOutput(graph: Graph, output: Output<TInt64>) : TensorFlowOutput<Long, TInt64>
Link copied to clipboard
abstract class TensorFlowAlgebra<T, TT : TNumber, A : Ring<T>> : TensorAlgebra<T, A>
Link copied to clipboard
value class TensorFlowArray<T>(tensor: NdArray<T>) : MutableStructureND<T>

Static (eager) in-memory TensorFlow tensor

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

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

Link copied to clipboard
interface TensorFlowTensor<T> : MutableStructureND<T>

Functions

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

Compute several outputs with TensorFlow in a single run.

Link copied to clipboard
fun DoubleField.produceWithTF(block: DoubleTensorFlowAlgebra.() -> StructureND<Double>): StructureND<Double>

Compute a tensor with TensorFlow in a single run.

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