init(splitCount:dimension:)
Creates a split layer with the number of splits and dimension you specify.
Declaration
convenience init(splitCount: Int, dimension: Int)Parameters
- splitCount:
The number of splits.
- dimension:
The dimension or axis along which to split the tensor.