提交 73f37330 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove ARange Op from JAX Subtensor tests

上级 e593b0ac
......@@ -601,7 +601,7 @@ def test_jax_scan_tap_output():
def test_jax_Subtensors():
# Basic indices
x_aet = aet.arange(3 * 4 * 5).reshape((3, 4, 5))
x_aet = aet.as_tensor(np.arange(3 * 4 * 5).reshape((3, 4, 5)))
out_aet = x_aet[1, 2, 0]
assert isinstance(out_aet.owner.op, aet_subtensor.Subtensor)
out_fg = FunctionGraph([], [out_aet])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论