split(_:splitSizesTensor:axis:name:)
Creates a split operation and returns the result tensor.
Declaration
func split(_ tensor: MPSGraphTensor, splitSizesTensor: MPSGraphTensor, axis: Int, name: String?) -> [MPSGraphTensor]Parameters
- tensor:
The input tensor
- splitSizesTensor:
The lengths of the result tensors along the split axis.
- axis:
The dimension along which MPSGraph splits the input tensor.
- name:
The name for the operation.
Return Value
A valid MPSGraphTensor object
Discussion
Splits the input tensor along axis into multiple result tensors of size determined by splitSizesTensor. Requires that the sum of the elements of splitSizesTensor is equal to the lenth of the input along axis.