Función Tensorflow.js tf.ones()

Tensorflow.js es una biblioteca de código abierto para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node. La función tf.ones() se usa para crear un nuevo tensor donde todos los elementos se establecen en 1.

Sintaxis:

tf.ones(shape, dtype, name)

Parámetros:

  • forma: Toma la forma del tensor. Puede ser una array multidimensional o un int32
  • dtype: Toma el tipo de dato del 1 que estamos insertando. Está configurado de forma predeterminada en float32, sin embargo, también puede ser int32. Es un parámetro opcional.
  • name: Toma el nombre de la operación que estamos realizando. Es Ninguno por defecto. Es un parámetro opcional.

Valor devuelto: Devuelve un tensor con el mismo tipo de dato.

Ejemplo 1: en este ejemplo, crearemos un tensor 2D con todos los valores establecidos en 1 de tipo de datos enteros.

JavaScript

// Import tensorflow
import tensorflow as tf
  
// Get a Tensor
val = tf.ones([10, 10], tf.int32)
  
// Print the Tensor
print(val)

Producción:

tf.Tensor(
[[1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]], shape=(10, 10), dtype=int32)

Ejemplo 2: en este ejemplo, crearemos un tensor 1D con todos los valores establecidos en 1 de tipo de datos flotante.

JavaScript

// Import tensorflow
import tensorflow as tf
  
// Get a Tensor
val = tf.ones(5, tf.float32)
  
// Print a Tensor
print(val)

Producción:

tf.Tensor([1. 1. 1. 1. 1.], shape=(5,), dtype=float32)

Referencia: https://js.tensorflow.org/api/latest/#ones

Publicación traducida automáticamente

Artículo escrito por pratikraut0000 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 *