variable_ops_test.py 文件源码

def testAssignUpdateNoVarShape(self):
    var = state_ops.variable_op([1, 2], tf.complex64, set_shape=False)
    added = tf.assign_add(var, [[2.0+2.0j, 3.0+3.0j]])
    self.assertEqual([1, 2], added.get_shape())
    subbed = tf.assign_sub(var, [[12.0+12.0j, 13.0+13.0j]])
    self.assertEqual([1, 2], subbed.get_shape())