wtf:wasserstein张量分解

时间:2024-04-26 03:22:27
【文件属性】:

文件名称:wtf:wasserstein张量分解

文件大小:900KB

文件格式:ZIP

更新时间:2024-04-26 03:22:27

Python

wtf:Wasserstein张量分解 Wasserstein损失平滑的非负矩阵和张量分解 该存储库包含文章“具有平滑Wasserstein损失的非负矩阵和张量分解”中描述的方法的基本实现。 要求 火炬 CUDA兼容的GPU(例如,使用 )可实现高效的autodiff 指示 克隆此仓库: git clone https://github.com/zsteve/wtf.git 引用 如果您发现这项工作与您的研究项目相关,请引用 Zhang, S. Non-negative matrix and tensor factorisations with a smoothed Wasserstein loss, arXiv preprint arXiv:2104.01708, 2021


【文件预览】:
wtf-main
----src()
--------wtf.py(11KB)
----images()
--------wtf.jpg(44KB)
----examples()
--------example.ipynb(1.14MB)
----LICENSE(1KB)
----README.md(809B)
----.gitignore(2KB)

网友评论