tf.raw_ops.TensorListConcat
Concats all tensors in the list along the 0th dimension.
tf.raw_ops.TensorListConcat( input_handle, element_dtype, element_shape=None, name=None )
Requires that all tensors have the same shape except the first dimension.
input_handle: The input list. tensor: The concated result. lengths: Output tensor containing sizes of the 0th dimension of tensors in the list, used for computing the gradient.
Args | |
---|---|
input_handle | A Tensor of type variant . |
element_dtype | A tf.DType . |
element_shape | An optional tf.TensorShape or list of ints . Defaults to None . |
name | A name for the operation (optional). |
Returns | |
---|---|
A tuple of Tensor objects (tensor, lengths). | |
tensor | A Tensor of type element_dtype . |
lengths | A Tensor of type int64 . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/TensorListConcat