BCELoss(reduction="sum") input = torch.randn(epoch, batch_size, requires_grad=True) target = torch.empty(epoch, batch_size, ).random_(2) ...
確定! 回上一頁