Contents

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) throws

Parameters

  • 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.

See Also

Activation layers