split(source:splitSectionLengths:dimension:)
Adds a new split layer to the graph using the source tensor, lengths of each split section, and dimension to split the source tensor that you specify.
Declaration
func split(source: MLCTensor, splitSectionLengths: [Int], dimension: Int) -> [MLCTensor]?Parameters
- source:
The source tensor.
- splitSectionLengths:
An array that contains the lengths of each split section.
- dimension:
The dimension along which to split the tensor.
Return Value
A result tensor.