Loss

class Loss(weight, batch_axis, **kwargs)

Base class for loss.

Parameters:

weight : float or None

Global scalar weight for loss.

batch_axis : int, default 0

The axis that represents mini-batch.