tensorflow-queue-0.1.0.0: Basic access to TensorFlow queues.

Safe HaskellNone
LanguageHaskell2010

TensorFlow.Queue

Description

Queues in TensorFlow graph. Very limited support for now.

Synopsis

Documentation

data Queue as

A queue carrying tuples.

makeQueue

Arguments

:: (MonadBuild m, TensorTypes as) 
=> Int64

The upper bound on the number of elements in this queue. Negative numbers mean no limit.

-> ByteString

If non-empty, this queue will be shared under the given name across multiple sessions.

-> m (Queue as) 

Creates a new queue with the given capacity and shared name.

enqueue :: forall as v m. (MonadBuild m, TensorTypes as) => Queue as -> TensorList v as -> m ControlNode

Adds the given values to the queue.

dequeue

Arguments

:: (MonadBuild m, TensorTypes as) 
=> Queue as 
-> m (TensorList Value as)

Dequeued tensors. They are coupled in a sense that values appear together, even if they are not consumed together.

Retrieves the values from the queue.