Tensor.Split¶
| 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 split the tensor. Default is 0. | 0 |
IntVar(Int)None |
Size |
No | The size of the sections to split the tensor into. | none |
IntVar(Int) |
Splits a tensor into multiple tensors along a specified dimension.