要将形状显示为整数列表,请执行tensor.get_shape().as_list()
。
要完成tf.shape()
通话,请尝试tensor2 = tf.reshape(tensor, tf.TensorShape([num_rows*num_cols, 1]))
。或者,您可以直接tensor2 = tf.reshape(tensor, tf.TensorShape([-1, 1]))
在可以推断出其第一维的地方进行操作。
如何获取Tensorflow张量尺寸(形状)作为int值?
要将形状显示为整数列表,请执行tensor.get_shape().as_list()
。
要完成tf.shape()
通话,请尝试tensor2 = tf.reshape(tensor, tf.TensorShape([num_rows*num_cols, 1]))
。或者,您可以直接tensor2 = tf.reshape(tensor, tf.TensorShape([-1, 1]))
在可以推断出其第一维的地方进行操作。