tf.raw_ops.TensorArraySplitV3  |  TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.TensorArraySplitV3

Stay organized with collections Save and categorize content based on your preferences.

Split the data from the input value into TensorArray elements.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.raw_ops.TensorArraySplitV3

tf.raw_ops.TensorArraySplitV3(
    handle, value, lengths, flow_in, name=None
)

Assuming that lengths takes on values

  (n0, n1, ..., n(T-1))

and that value has shape

  (n0 + n1 + ... + n(T-1) x d0 x d1 x ...),

this splits values into a TensorArray with T tensors.

TensorArray index t will be the subtensor of values with starting position

  (n0 + n1 + ... + n(t-1), 0, 0, ...)

and having size

  nt x d0 x d1 x ...
Args
handle A Tensor of type resource. The handle to a TensorArray.
value A Tensor. The concatenated tensor to write to the TensorArray.
lengths A Tensor of type int64. The vector of lengths, how to split the rows of value into the TensorArray.
flow_in A Tensor of type float32. A float scalar that enforces proper chaining of operations.
name A name for the operation (optional).
Returns
A Tensor of type float32.