init(function:i_desc:o_desc:reduction:huber_delta:)
Returns a new Huber loss layer parameters structure from the specified parameters.
Declaration
init(function: BNNSLossFunction, i_desc: BNNSNDArrayDescriptor, o_desc: BNNSNDArrayDescriptor, reduction: BNNSLossReductionFunction, huber_delta: 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.
- huber_delta:
The boundary value that defines where Huber loss returns mean absolute error or mean square error.