ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Stage (key, values) in the underlying container which behaves like a ordered

associative container. Elements are ordered by key.

key A Tensor of type int64. int64
indices A Tensor of type int32.
values A list of Tensor objects. a list of tensors dtypes A list of data types that inserted values should adhere to.
dtypes A list of tf.DTypes.
capacity An optional int that is >= 0. Defaults to 0. Maximum number of elements in the Staging Area. If > 0, inserts on the container will block when the capacity is reached.
memory_limit An optional int that is >= 0. Defaults to 0.
container An optional string. Defaults to "". If non-empty, this queue is placed in the given container. Otherwise, a default container is used.
shared_name An optional string. Defaults to "". It is necessary to match this name to the matching Unstage Op.
name A name for the operation (optional).

The created Operation.