SoftmaxOperator

public class SoftmaxOperator: ActivationOperator

Softmax activations.

y = exp(x) / reduce_sum(exp(x), dim)

Default dim is the last dimension of input tensors.

Note

The operator assumens all input tensors have same rank. If not, it could not do the calculation.


                    
                    
                    init(computationDelegate:)

Convenience initializer Subclass required to override this function to assign cpuComputeBlock and metalKernelFuncLabel

Swift

required public convenience init(computationDelegate: OperatorCalculationDelegate? = nil)


                                computationDelegate

computationDelegate

Show on GitHub


                    
                    
                    init(computationDelegate:dim:)

Initial by setting dim value.

Swift

public convenience init(computationDelegate: OperatorCalculationDelegate? = nil, dim: Int)

`computationDelegate`	computationDelegate description
`dim`	reduce dim

Show on GitHub


                    
                    
                    outputShape(shapeArray:)

Override this function from UnaryOperaotr to do additional checking validation bewtween dim and shapes.

Swift

public override func outputShape(shapeArray shapes: [TensorShape]) -> [TensorShape]?


                                shapes

shapes description

return value description

Show on GitHub


                    
                    
                    compute(_:)

Override computaion methods. This computation just call other operators.

Swift

public override func compute(_ computationMode: OperatorComputationMode = SerranoEngine.configuredEngine.defaultComputationMode)


                                computationMode

computationMode

Show on GitHub