Solución:
Para obtener la forma como una lista de enteros, haga tensor.get_shape().as_list()
.
Para completar tu tf.shape()
llamar, intentar tensor2 = tf.reshape(tensor, tf.TensorShape([num_rows*num_cols, 1]))
. O puedes hacer directamente tensor2 = tf.reshape(tensor, tf.TensorShape([-1, 1]))
donde se puede inferir su primera dimensión.
Otra forma de resolver esto es así:
tensor_shape[0].value
Esto devolverá el valor int del objeto Dimension.
para un tensor 2-D, puede obtener el número de filas y columnas como int32 usando el siguiente código:
rows, columns = map(lambda i: i.value, tensor.get_shape())
¡Haz clic para puntuar esta entrada!
(Votos: 0 Promedio: 0)