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

TensorFlowOutput

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

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

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

Inheritors

DoubleTensorFlowOutput

IntTensorFlowOutput

LongTensorFlowOutput

Constructors

TensorFlowOutput

constructor(graph: Graph, output: Output<TT>)

Functions

@PerformancePitfall

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

@PerformancePitfall

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

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

@PerformancePitfall

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: ShapeND