split(source:splitCount:dimension:)
Adds a new split layer to the graph using the source tensor, number of splits, and dimension to split the source tensor that you specify.
Declaration
func split(source: MLCTensor, splitCount: Int, dimension: Int) -> [MLCTensor]?Parameters
- source:
The source tensor.
- splitCount:
The number of splits.
- dimension:
The dimension along which to split the tensor.
Return Value
A result tensor.