Contents

init(splitSectionLengths:dimension:)

Creates a split layer with the lengths of each split section and dimension you specify.

Declaration

convenience init(splitSectionLengths: [Int], dimension: Int)

Parameters

  • splitSectionLengths:

    An array that contains the lengths of each split section.

  • dimension:

    The dimension or axis along which to split the tensor.

See Also

Creating Split Layers