variable_ops_test.py 文件源码

def testAssignNoVarShapeNoValidateShape(self):
    value = np.array([[42.0+42.0j, 43.0+43.0j]])
    var = state_ops.variable_op(value.shape, tf.complex64, set_shape=False)
    self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
    assigned = tf.assign(var, value, validate_shape=False)
    self.assertShapeEqual(value, assigned)