def sigmoid(x): return 1/(1+math.exp(-x)) output = [sigmoid(x) for x in output] ... 可以看到, PyTorch 封裝好的BCELoss 應該是有四捨五入的。
確定! 回上一頁