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 Queue2 a b Source

A queue carrying tuples. The underlying structure is more versatile and can be made to support arbitrary tuples.

makeQueue2 Source

Arguments

:: (TensorType a, TensorType b) 
=> 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.

-> Build (Queue2 a b) 

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

enqueue :: forall a b v1 v2. (TensorType a, TensorType b) => Queue2 a b -> Tensor v1 a -> Tensor v2 b -> Build ControlNode Source

Adds the given values to the queue.

dequeue Source

Arguments

:: (TensorType a, TensorType b) 
=> Queue2 a b 
-> Build (Tensor Ref a, Tensor Ref b)

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

Retrieves the values from the queue.