参考链接需要的包import torchimport torch.nn as nnimport torch.nn.functional ... y): return -(pred.log()*y + (1-y)*(1-pred).log()).mean() pred = sigmoid(x) ...
確定! 回上一頁