Tensor.Stack¶
| Name | Mandatory | Description | Default | Type |
|---|---|---|---|---|
⬅️ Input |
The input of the shard, if any | [Tensor] |
||
Output ➡️ |
The resulting output of the shard | Tensor |
||
Dim |
No | The dimension along which to stack the tensors. Default is 0. | 0 |
IntVar(Int)None |
Stacks a sequence of tensors along a new axis.