We need a new loss class for multi-class classification: SoftmaxCrossEntropy . The class should inherit from…
We need a new loss class for multi-class classification: SoftmaxCrossEntropy . The class should inherit from…: a task in LegoFlow-SWE (Harbor dataset). Forward method: - Inputs: - prediction – a 2D tensor of shape batch size, num classes containing logits (raw scores) for each class. - target – a…
The task
**Forward** method: - Inputs: - `prediction` – a 2D tensor of shape `[batch_size, num_classes]` containing logits (raw scores) for each class. - `target` – a 1D tensor of shape `[batch_size, 1]` holding ground-truth class indices as integers (e.g., `kInt` type). - For each sample, apply the **SoftMax** function to…
Part of Lego-X/LegoFlow-SWE.