【文件属性】：
文件名称：PyTorchOT:在pytorch中实现最佳传输算法
文件大小：22.06MB
文件格式：ZIP
更新时间：2021-05-24 13:26:27
Python
PyTorchOT
在PyTorch中实现接收器优化传输算法。 目前，已实现了Sinkhorn算法的两个版本：和对。 这段代码实际上只是从PyTorch中很棒的POT库（ ）中重新实现了一些实现。
用法示例：
from ot_pytorch import sink
M = pairwise_distance_matrix()
dist = sink(M, reg=5, cuda=False)
设置cuda = True启用cuda使用。
examples.py文件包含两个基本示例。
范例1：
让Z I〜统一[0,1]，并定义数据X i =（0，Z I）中，Y I =（θ，Z i）中，对于i = 1，...，N和一些参数θ，其在[-1,1]上变化。 真正的最佳传输距离是|θ|。 该算法产生：
【文件预览】：
PyTorchOT-master
----examples.py(3KB)
----plots()
--------uniform_example()
--------gaussian_example()
----README.md(1000B)
----ot_pytorch.py(5KB)
----data()
--------raw()
--------uniform_example()
--------processed()
--------gaussian_example()