Skip to content
#

wasserstein

Star

Here are 24 public repositories matching this topic...

Language: All
Filter by language
All 24 Python 11 Jupyter Notebook 7 C++ 2 Lua 1 MATLAB 1
Sort: Best match
Sort options
Best match Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

PythonOT / POT

Star
Open

ot.gpu.sinkhorn uses dtype of the cost matrix

2
RaitoBezarius
RaitoBezarius commented Aug 26, 2019

I tried to use ot.gpu.sinkhorn using CUDA and I got this traceback:

  File "/lib/python3.5/site-packages/ot/gpu/bregman.py", line 132, in sinkhorn_knopp
    np.divide(M, -reg, out=K)
  File "cupy/core/_kernel.pyx", line 831, in cupy.core._kernel.ufunc.__call__
  File "cupy/core/_kernel.pyx", line 355, in cupy.core._kernel._get_out_args
TypeError: output (typecode 'd') could not be c
Read more
documentation

Improve this page

Add a description, image, and links to the wasserstein topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the wasserstein topic, visit your repo's landing page and select "manage topics."

Learn more

You can’t perform that action at this time.