'mean': elemenwise mean, 'sum': class dim will be summed and batch dim will be averaged. use_weight : whether to use weights of classes. weight : Tensor, ...
確定! 回上一頁