文件名称：PyTorchOT:在pytorch中实现最佳传输算法
文件大小：22.06MB
文件格式：ZIP
更新时间：2024-06-13 07:13:07
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()