init(operation:dimension:)
Creates a softmax layer with the operation and dimension you specify.
Declaration
convenience init(operation: MLCSoftmaxOperation, dimension: Int)Parameters
- operation:
The softmax operation.
- dimension:
The dimension over which you perform the softmax operation.