Skip to content
Snippets Groups Projects
Unverified Commit ac06d3c8 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

remove useless code in expand module test (#4903)

parent 698e8cd5
No related branches found
No related tags found
No related merge requests found
......@@ -118,24 +118,6 @@ class TestModule(flow.unittest.TestCase):
test_case.assertTrue(np.array_equal(of_out.numpy(), out_np.astype(np.int32)))
def test_expand_same_int(test_case):
input_shape = (2, 4, 1, 32)
expand_dim = [2, 4, 2, 32]
input, gout, out_np, gin_np = getExpandGrad(input_shape, expand_dim)
of_input = flow.Tensor(input, dtype=flow.int)
of_out = of_input.expand(expand_size=expand_dim)
test_case.assertTrue(np.array_equal(of_out.numpy(), out_np.astype(np.int32)))
def test_expand_same_int(test_case):
input_shape = (2, 4, 1, 32)
expand_dim = [2, 4, 2, 32]
input, gout, out_np, gin_np = getExpandGrad(input_shape, expand_dim)
of_input = flow.Tensor(input, dtype=flow.int)
of_out = of_input.expand(expand_size=expand_dim)
test_case.assertTrue(np.array_equal(of_out.numpy(), out_np.astype(np.int32)))
if __name__ == "__main__":
unittest.main()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment