MCPcopy Create free account
hub / github.com/PythonOT/POT / allclose

Method allclose

ot/backend.py:3213–3214  ·  view source on GitHub ↗
(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False)

Source from the content-addressed store, hash-verified

3211 return tf.identity(a)
3212
3213 def allclose(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False):
3214 return tnp.allclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan)
3215
3216 def dtype_device(self, a):
3217 return a.dtype, a.device.split("device:")[1]

Callers 1

test_gradients_backendsFunction · 0.95

Calls 1

allcloseMethod · 0.45

Tested by 1

test_gradients_backendsFunction · 0.76