PoissonLoss

constructor(logInput: Boolean = true, epsilon: Float = 1.0E-8f)(source)

Parameters

logInput

If true, the input is in log-space (e.g., output of a linear layer) and will be exponentiated. If false, input should already be positive (e.g., output of softplus). Default is true.

epsilon

Small value for numerical stability when logInput is false. Default is 1e-8.