open3d.ml.torch.modules.losses.CrossEntropyLoss#

class open3d.ml.torch.modules.losses.CrossEntropyLoss(loss_weight=1.0)#

CrossEntropyLoss.

__init__(loss_weight=1.0)#

CrossEntropyLoss.

Parameters:

loss_weight (float, optional) – Weight of the loss. Defaults to 1.0.

forward(cls_score, label, weight=None, avg_factor=None, **kwargs)#

Forward function.

Parameters:
  • cls_score (torch.Tensor) – The prediction.

  • label (torch.Tensor) – The learning label of the prediction.

  • weight (torch.Tensor, optional) – Sample-wise loss weight.

  • avg_factor (int, optional) – Average factor that is used to average the loss. Defaults to None.

Returns:

The calculated loss

Return type:

torch.Tensor