提交 88f6d2db authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Change all += to np.add.at()

上级 83b6a590
...@@ -118,7 +118,7 @@ def test_advinc_subtensor1_dtype(): ...@@ -118,7 +118,7 @@ def test_advinc_subtensor1_dtype():
for node in f.maker.fgraph.toposort()]) == 1 for node in f.maker.fgraph.toposort()]) == 1
rval = f(yval) rval = f(yval)
rep = xval.copy() rep = xval.copy()
rep[[0, 2]] += yval np.add.at(rep, [[0, 2]], yval)
assert np.allclose(rval, rep) assert np.allclose(rval, rep)
...@@ -140,7 +140,7 @@ def test_deterministic_flag(): ...@@ -140,7 +140,7 @@ def test_deterministic_flag():
for node in f.maker.fgraph.toposort()]) == 1 for node in f.maker.fgraph.toposort()]) == 1
rval = f(yval) rval = f(yval)
rep = xval.copy() rep = xval.copy()
rep[[0, 2]] += yval np.add.at(rep, [[0, 2]], yval)
assert np.allclose(rval, rep) assert np.allclose(rval, rep)
...@@ -184,7 +184,7 @@ def test_incsub_f16(): ...@@ -184,7 +184,7 @@ def test_incsub_f16():
for node in f.maker.fgraph.toposort()]) == 1 for node in f.maker.fgraph.toposort()]) == 1
rval = f(yval) rval = f(yval)
rep = xval.copy() rep = xval.copy()
rep[[0, 2]] += yval np.add.at(rep, [[0, 2]], yval)
assert np.allclose(rval, rep) assert np.allclose(rval, rep)
expr = tensor.inc_subtensor(x[1:], y) expr = tensor.inc_subtensor(x[1:], y)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论