@Operator public final class TensorListSplit extends PrimitiveOp implements Operand<Object>
list[i] corresponds to lengths[i] tensors from the input tensor. The tensor must have rank at least 1 and contain exactly sum(lengths) elements.
tensor: The input tensor. element_shape: A shape compatible with that of elements in the tensor. lengths: Vector of sizes of the 0th dimension of tensors in the list. output_handle: The list.
operation
Modifier and Type | Method and Description |
---|---|
Output<Object> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T,U extends Number> |
create(Scope scope,
Operand<T> tensor,
Operand<U> elementShape,
Operand<Long> lengths)
Factory method to create a class wrapping a new TensorListSplit operation.
|
Output<?> |
outputHandle() |
equals, hashCode, op, toString
public static <T,U extends Number> TensorListSplit create(Scope scope, Operand<T> tensor, Operand<U> elementShape, Operand<Long> lengths)
scope
- current scopetensor
- elementShape
- lengths
- public Output<?> outputHandle()
public Output<Object> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<Object>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.