Unpacks a given dimension of a rank-`R` tensor into `num` rank-`(R-1)` tensors.
Unpacks `num` tensors from `value` by chipping it along the `axis` dimension. For example, given a tensor of shape `(A, B, C, D)`;
If `axis == 0` then the i'th tensor in `output` is the slice `value[i, :, :, :]` and each tensor in `output` will have shape `(B, C, D)`. (Note that the dimension unpacked along is gone, unlike `split`).
If `axis == 1` then the i'th tensor in `output` is the slice `value[:, i, :, :]` and each tensor in `output` will have shape `(A, C, D)`. Etc.
This is the opposite of `pack`.
Nested Classes
class | Unstack.Options | Optional attributes for Unstack
|
Public Methods
static Unstack.Options |
axis(Long axis)
|
static <T> Unstack<T> |
create(Scope scope, Operand<T> value, Long num, Options... options)
Factory method to create a class wrapping a new Unstack operation.
|
Iterator<Operand<T>> |
iterator()
|
List<Output<T>> |
output()
The list of tensors unpacked from `value`.
|
Inherited Methods
Public Methods
public static Unstack.Options axis (Long axis)
Parameters
axis | Dimension along which to unpack. Negative values wrap around, so the valid range is `[-R, R)`. |
---|
public static Unstack<T> create (Scope scope, Operand<T> value, Long num, Options... options)
Factory method to create a class wrapping a new Unstack operation.
Parameters
scope | current scope |
---|---|
value | 1-D or higher, with `axis` dimension size equal to `num`. |
options | carries optional attributes values |
Returns
- a new instance of Unstack