def testConstantInvalidValue(self): c = constant_op.constant([1.0, 2.0, 3.0]) with self.assertRaisesRegex(TypeError, r"Invalid type for initial value: .*Tensor.*"): init_ops_v2.constant_initializer(c) v = variables.Variable([3.0, 2.0, 1.0]) with self.assertRaisesRegex( TypeError, r"Invalid type for initial value: .*Variable.*"): init_ops_v2.constant_initializer(v)
def testConstantInvalidValue(self): c = constant_op.constant([1.0, 2.0, 3.0]) with self.assertRaisesRegexp( TypeError, r"Invalid type for initial value: .*Tensor.*"): init_ops_v2.constant_initializer(c) v = variables.Variable([3.0, 2.0, 1.0]) with self.assertRaisesRegexp( TypeError, r"Invalid type for initial value: .*Variable.*"): init_ops_v2.constant_initializer(v)
def _testNDimConstantInitializer(self, value, shape, expected): with test_util.use_gpu(): init = init_ops_v2.constant_initializer(value) x = init(shape) actual = self.evaluate(array_ops.reshape(x, [-1])) self.assertEqual(len(actual), len(expected)) for a, e in zip(actual, expected): self.assertEqual(a, e)
def _testNDimConstantInitializerIncorrectNumberValues(self, value, shape): with test_util.use_gpu(): init = init_ops_v2.constant_initializer(value) self.assertRaises(TypeError, init, shape=shape)
def testConstantTuple(self): init = init_ops_v2.constant_initializer((10, 20, 30)) tensor = init(shape=[3]) self.assertAllEqual(self.evaluate(tensor), [10, 20, 30]) self.assertEqual(tensor.shape, [3])