在PyTorch中沿轴上的所有索引应用函数我试图在PyTorch中实现Wasserstein Loss函数,为此我引用了Scipy实现。因为在forward()方法中使用PyTorch函数意味着不必编写backward()函数,所以我在代码中 ...2024-10-06 已阅读: n次