TensorFlow: cómo crear un tensor de todos los que tiene la misma forma que el tensor de entrada

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.

Método utilizado:

  • ones_like: este método acepta un Tensor como entrada y devuelve un Tensor con la misma forma que tiene todos los valores establecidos en uno.

Ejemplo 1:

Python3

# importing the library
import tensorflow as tf
  
# Initializing the Input
input = tf.constant([[1, 2, 3], [4, 5, 6]])
  
# Generating Tensor with having all values as 1
res = tf.ones_like(input)
  
# Printing the resulting Tensors
print("Res: ", res )

Producción:

Res:  tf.Tensor(
[[1 1 1]
 [1 1 1]], shape=(2, 3), dtype=int32)

Ejemplo 2: este ejemplo especifica explícitamente el tipo del tensor resultante.

Python3

# importing the library
import tensorflow as tf
  
# Initializing the Input
input = tf.constant([[1, 2, 3], [4, 5, 6]])
  
# Printing the Input
print("Input: ", input)
  
# Generating Tensor with having all values as 1
res = tf.ones_like(input, dtype = tf.float64)
  
# Printing the resulting Tensors
print("Res: ", res )

Producción:

Input:  tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
Res:  tf.Tensor(
[[1. 1. 1.]
 [1. 1. 1.]], shape=(2, 3), dtype=float64)


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

Deja una respuesta

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