quote form bceloss 1.



weight (Tensor, optional) – a manual rescaling weight given to the loss of each batch element. If given, has to be a Tensor of size nbatch.


do you want to use this feature or you want to have weight for each class?