标签： python tensorflow keras pytorch
我刚开始使用Pytorch。我有两组观测数据Y和X，可能具有不同的维度。我的任务是训练函数g，以使g(X)和Y之间的分布距离最小。我想将Wasserstein距离作为损失函数。是否可以使用Pytorch中的Wasserstein距离自定义损失函数？如果没有，有可能用Keras写作吗？如果两种方法都不起作用，那应该怎么办呢？
Y
X
g
g(X)