variable_ops_test.py 文件源码

def testset_shape(self):
    p = state_ops.variable_op([1, 2], tf.complex64)
    self.assertEqual([1, 2], p.get_shape())
    p = state_ops.variable_op([1, 2], tf.complex64, set_shape=False)
    self.assertEqual(tensor_shape.unknown_shape(), p.get_shape())