Python _get_size Examples

Example #1

File: test_generator.py Project: ZhangAustin/chainer

    def check_normal(self, curand_func, dtype):
        shape = cupy._get_size(self.size)
        exp_size = six.moves.reduce(operator.mul, shape, 1)
        if exp_size % 2 == 1:
            exp_size += 1

        curand_func.return_value = cupy.zeros(exp_size, dtype=dtype)
        out = self.rs.normal(self.args[0], self.args[1], self.size, dtype)
        gen, _, size, loc, scale = curand_func.call_args[0]
        self.assertIs(gen, self.rs._generator)
        self.assertEqual(size, exp_size)
        self.assertIs(loc, self.args[0])
        self.assertIs(scale, self.args[1])
        self.assertEqual(out.shape, shape)

Example #2

File: test_generator.py Project: hokuto-k/chainer

    def check_normal(self, curand_func, dtype):
        shape = cupy._get_size(self.size)
        exp_size = six.moves.reduce(operator.mul, shape, 1)
        if exp_size % 2 == 1:
            exp_size += 1

        curand_func.return_value = cupy.zeros(exp_size, dtype=dtype)
        out = self.rs.normal(self.args[0], self.args[1], self.size, dtype)
        gen, _, size, loc, scale = curand_func.call_args[0]
        self.assertIs(gen, self.rs._generator)
        self.assertEqual(size, exp_size)
        self.assertIs(loc, self.args[0])
        self.assertIs(scale, self.args[1])
        self.assertEqual(out.shape, shape)

Example #3

 def _generate_normal(self, func, size, dtype, *args):
     # curand funcitons below don't support odd size.
     # * curand.generateNormal
     # * curand.generateNormalDouble
     # * curand.generateLogNormal
     # * curand.generateLogNormalDouble
     size = cupy._get_size(size)
     element_size = six.moves.reduce(operator.mul, size, 1)
     if element_size % 2 == 0:
         out = cupy.empty(size, dtype=dtype)
         func(self._generator, out.data.ptr, out.size, *args)
         return out
     else:
         out = cupy.empty((element_size + 1,), dtype=dtype)
         func(self._generator, out.data.ptr, out.size, *args)
         return out[:element_size].reshape(size)

Example #4

File: generator.py Project: himawarijp/chainer

 def _generate_normal(self, func, size, dtype, *args):
     # curand funcitons below don't support odd size.
     # * curand.generateNormal
     # * curand.generateNormalDouble
     # * curand.generateLogNormal
     # * curand.generateLogNormalDouble
     size = cupy._get_size(size)
     element_size = six.moves.reduce(operator.mul, size, 1)
     if element_size % 2 == 0:
         out = cupy.empty(size, dtype=dtype)
         func(self._generator, out.data.ptr, out.size, *args)
         return out
     else:
         out = cupy.empty((element_size + 1,), dtype=dtype)
         func(self._generator, out.data.ptr, out.size, *args)
         return out[:element_size].reshape(size)

Example #5

 def check_collection(self, a):
     self.assertEqual(cupy._get_size(a), tuple(a))

Example #6

 def test_none(self):
     self.assertEqual(cupy._get_size(None), ())

Example #7

 def test_float(self):
     with self.assertRaises(ValueError):
         cupy._get_size(1.0)

Example #8

 def test_int(self):
     self.assertEqual(cupy._get_size(1), (1, ))

Example #9

File: test_ndarray.py Project: k-kato-morpho/chainer

 def test_none(self):
     self.assertEqual(cupy._get_size(None), ())

Example #10

File: test_ndarray.py Project: k-kato-morpho/chainer

 def test_float(self):
     with self.assertRaises(ValueError):
         cupy._get_size(1.0)

Example #11

File: test_ndarray.py Project: k-kato-morpho/chainer

 def test_int(self):
     self.assertEqual(cupy._get_size(1), (1,))

Example #12

File: test_ndarray.py Project: k-kato-morpho/chainer

 def check_collection(self, a):
     self.assertEqual(cupy._get_size(a), tuple(a))