Saltar al contenido

¿Cómo obtener las dimensiones (forma) del tensor de Tensorflow como valores int?

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)



Utiliza Nuestro Buscador

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *