Lines Matching refs:numpy
21 import numpy as np
41 got.append(t.numpy())
46 self.assertEqual(0, iterator.get_next().numpy())
47 self.assertEqual(1, iterator.get_next().numpy())
48 self.assertEqual(2, iterator.get_next().numpy())
49 self.assertEqual(3, iterator.get_next().numpy())
57 got = [x.numpy() for x in it1]
60 got = [x.numpy() for x in it2]
71 got = (x[0].numpy(), (x[1][0].numpy(), x[1][1].numpy()))
81 got = [x.numpy() for x in it]
93 got = [x.numpy() for x in it]
99 got1 = [x.numpy() for x in datasets.Iterator(ds)]
101 got2 = [x.numpy() for x in datasets.Iterator(ds)]
158 got = [x.numpy() for x in datasets.Iterator(ds)]
166 self.assertAllEqual([0., 2.], x.numpy())
