Added another test

上级 a30ce194
......@@ -463,6 +463,16 @@ class T_log2(unittest.TestCase):
def test0(self):
verify_grad(self, Log2, [numpy.random.rand(3,1)+0.0001])
class T_log(unittest.TestCase):
def test0(self):
verify_grad(self, Log, [numpy.random.rand(3,1)+0.0001])
def test1(self):
a = astensor(numpy.ones(2))
b = astensor(numpy.ones(2))
aa = numpy.asarray([0.5, 4.0])
bb = numpy.asarray([0.5, 2.0])
check_eq2(self, [a], log(a), [aa], numpy.log(numpy.asarray(aa)))
class T_pow(unittest.TestCase):
def setUp(self):
numpy.random.seed(9999)
......
......@@ -81,7 +81,10 @@ def grad_sources_inputs(sources, graph_inputs):
op_grad = op.grad(input_arg, output_arg)
if op_grad is None:
raise ValueError(_msg_retNone, op.__class__)
if isinstance(op_grad, float):
raise TypeError('wtf!!!!!!!!', op)
g_inputs = _pack_result(op_grad)
assert isinstance(g_inputs, (list, tuple))
if len(g_inputs) != len(op.inputs):
raise ValueError(_msg_badlen,
op.__class__,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论