TensorFlow es una biblioteca Python de código abierto diseñada por Google para desarrollar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo.
sure_shape() se usa para actualizar y verificar la forma de Tensor.
Sintaxis: tensorflow.ensure_shape(x, forma, nombre)
Parámetros:
- x: Es Tensor de entrada.
- forma: es TensorShape que representa la forma del tensor de entrada.
- name(opcional): Define el nombre de la operación.
Devoluciones:
Devuelve un Tensor igual que x o genera tf.errors.InvalidArgumentError si las formas son incompatibles.
Ejemplo 1:
Python3
# Importing the library import tensorflow as tf # Initializing the input x = tf.constant([[2, 3, 6], [4, 8, 15]]) # Printing the input print('x:', x) # Calculating result res = tf.ensure_shape(x, (2, 3)) # Printing the result print('res: ', res)
Producción:
x: tf.Tensor( [[ 2 3 6] [ 4 8 15]], shape=(2, 3), dtype=int32) res: tf.Tensor( [[ 2 3 6] [ 4 8 15]], shape=(2, 3), dtype=int32)
Ejemplo 2: en este ejemplo, la forma es incompatible con la forma de x, por lo que se genera un error.
Python3
# Importing the library import tensorflow as tf # Initializing the input x = tf.constant([[2, 3, 6], [4, 8, 15]]) # Printing the input print('x:', x) # Calculating result res = tf.ensure_shape(x, (2, 4)) # Printing the result print('res: ', res)
Producción:
x: tf.Tensor( [[ 2 3 6] [ 4 8 15]], shape=(2, 3), dtype=int32) --------------------------------------------------------------------------- InvalidArgumentError Traceback (most recent call last) <ipython-input-1-ab1be364fadb> in <module>() 9 10 # Calculating result ---> 11 res = tf.ensure_shape(x, (2, 4)) 12 13 # Printing the result 3 frames /usr/local/lib/python3.6/dist-packages/six.py in raise_from(value, from_value) InvalidArgumentError: Shape of tensor dummy_input [2, 3] is not compatible with expected shape [2, 4]. [Op:EnsureShape]
Publicación traducida automáticamente
Artículo escrito por aman neekhara y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA