Python infer_unknown_dimension Examples

Example #1

File: test_internal.py Project: thomasaarholt/cupy

 def test_infer(self):
     assert internal.infer_unknown_dimension((-1, 2, 3), 12) == [2, 2, 3]

Example #2

File: test_internal.py Project: thomasaarholt/cupy

 def test_known_all(self):
     assert internal.infer_unknown_dimension((1, 2, 3), 6) == [1, 2, 3]

Example #3

File: test_internal.py Project: thomasaarholt/cupy

 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)

Example #4

File: test_internal.py Project: Lynkzhang/Cupy-UM

 def test_infer(self):
     self.assertEqual(internal.infer_unknown_dimension((-1, 2, 3), 12),
                      [2, 2, 3])

Example #5

File: test_internal.py Project: Lynkzhang/Cupy-UM

 def test_known_all(self):
     self.assertEqual(internal.infer_unknown_dimension((1, 2, 3), 6),
                      [1, 2, 3])

Example #6

File: test_internal.py Project: owruby/chainer

 def test_infer(self):
     self.assertEqual(internal.infer_unknown_dimension((-1, 2, 3), 12),
                      [2, 2, 3])

Example #7

File: test_internal.py Project: owruby/chainer

 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)

Example #8

File: test_internal.py Project: owruby/chainer

 def test_known_all(self):
     self.assertEqual(internal.infer_unknown_dimension((1, 2, 3), 6),
                      [1, 2, 3])