By argsorting, the highest value along the given dimension will get the value 0 tmp = torch.argsort(y_hat, dim=1) # Then compare to 0, ...
確定! 回上一頁