kmath-tensorflow/space.kscience.kmath.tensorflow/TensorFlowOutput

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.

If the tensor is used for intermediate operations, actualizing it could impact performance.

Constructors

TensorFlowOutput

fun <TT : TType> TensorFlowOutput(graph: Graph, output: Output<TT>)

Functions

@PerformancePitfall

open override fun elements(): Sequence<Pair<IntArray, T>>

open operator override fun get(index: IntArray): T

open override fun <F : StructureFeature> getFeature(type: KClass<out F>): F?

open operator override fun set(index: IntArray, value: T)

Properties

open val dimension: Int

open val indices: ShapeIndexer

var output: Output<TT>

open override val shape: Shape

Inheritors

DoubleTensorFlowOutput

IntTensorFlowOutput

LongTensorFlowOutput

Sources