PyTorch svd（）重建失败

Question

这一点已在评论中得到解决

---

当使用nn.DataPrallel时，以下测试都返回False：

（1）使用torch.svd()

u_, s_, v_ = torch.svd(x)
is_same = torch.allclose(x, torch.matmul(torch.matmul(u_, torch.diag_embed(s_)), v_.transpose(-2, -1)))

（2）使用torch.linalg.svd()

u_, s_, vh_ = torch.linalg.svd(x)
is_same = torch.allclose(x, u_ @ torch.diag_embed(s_) @ vh_)

当不使用nn.DataParallel时，它们返回True。 x是大小为非负值（22,64,64）的张量