Contents

init(function:i_desc:o_desc:reduction:label_smooth:)

Returns a new softmax cross entropy loss layer parameters structure from the specified parameters.

Declaration

init(function: BNNSLossFunction, i_desc: BNNSNDArrayDescriptor, o_desc: BNNSNDArrayDescriptor, reduction: BNNSLossReductionFunction, label_smooth: Float)

Parameters

  • function:

    The function that’s used to compute loss.

  • i_desc:

    The descriptor of the input.

  • o_desc:

    The descriptor of the output.

  • reduction:

    The function that’s used to reduce the computed loss.

  • label_smooth:

    A value that defines the smoothing that the loss function applies to the labels.

Discussion

See Also

Initializers