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())
评论列表
文章目录