applyActivation(activation:axes:input:output:batchSize:filterParameters:)
Applies an activation function on the specified axes.
Declaration
static func applyActivation(activation: BNNS.ActivationFunction, axes: [Int], input: BNNSNDArrayDescriptor, output: BNNSNDArrayDescriptor, batchSize: Int, filterParameters: BNNSFilterParameters? = nil) throwsParameters
- activation:
The activation that the function applies.
- axes:
The indices of the axes on which the function applies certain activation functions, such as Softmax.
- input:
The descriptor of the input.
- output:
The descriptor of the output.
- batchSize:
The number of input-output pairs.
- filterParameters:
The filter runtime parameters.