dmlc--dgl
Minjie Wang
8ea359d125
* add examples in traversal.py * message propagate methods * use the new message propagation for tree-lstm * update to the new name * update propagate API doc * update doc * add propagate utest
12 行
219 B
Python
12 行
219 B
Python
import torch as th
|
|
|
|
def allclose(a, b):
|
|
return th.allclose(a, b, rtol=1e-4, atol=1e-4)
|
|
|
|
def check_fail(fn, *args, **kwargs):
|
|
try:
|
|
fn(*args, **kwargs)
|
|
return False
|
|
except:
|
|
return True
|