... batch_size) rk = torch.argsort(loss_vector, descending=True) loss = loss_vector[rk[:n]].mean() return loss def chisq(loss_vector, alpha, ...
確定! 回上一頁