variable_ops_test.py 文件源码

def testAssignUpdateNoValueShape(self):
    var = state_ops.variable_op([1, 2], tf.complex64)
    added = tf.assign_add(var, self._NewShapelessTensor())
    self.assertEqual([1, 2], added.get_shape())
    subbed = tf.assign_sub(var, self._NewShapelessTensor())
    self.assertEqual([1, 2], subbed.get_shape())