You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
input (Tensor): Input tensor, the data type is float32 or float64. Shape is (N, C), where C is number of classes, and if shape is more than 2D, this is (N, C, D1, D2,..., Dk), k >= 1.
3459
3459
label (Tensor): Label tensor, the data type is float32 or float64. The shape of label is the same as the shape of input.
3460
-
weight (Tensor,optional): a manual rescaling weight given to each class.
3460
+
weight (Tensor,optional): a manual rescaling weight given to each class.
3461
3461
If given, has to be a Tensor of size C and the data type is float32, float64.
3462
3462
Default is ``'None'`` .
3463
3463
reduction (str, optional): Indicate how to average the loss by batch_size,
input: N-D Tensor, the shape is [N, \*], N is batch size and `\*` means number of classes, available dtype is float32, float64. The sum operation operates over all the elements.
3474
3474
label: N-D Tensor, same shape as the input.
3475
-
weight:N-D Tensor, the shape is [N,1]
3475
+
weight:N-D Tensor, the shape is [N,1]
3476
3476
output: scalar. If :attr:`reduction` is ``'none'``, then same shape as the input.
0 commit comments