Función Tensorflow.js tf.cast()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google 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.cast() se utiliza para convertir un tensor especificado en un nuevo tipo de datos.

Sintaxis:

tf.cast (x, dtype)

Parámetros: Esta función acepta dos parámetros que se ilustran a continuación:

  • x: El tensor de entrada que se está proyectando.
  • dtype: el tipo de datos en el que se va a convertir el tensor de entrada.

Valor de retorno: Devuelve un tensor moldeado de nuevo tipo de datos.

Ejemplo 1:

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor of some values
const x = tf.tensor1d([2.3, 1.7, 5, 0, 1, 0.5]);
  
// Calling the .cast() function over the 
// above tensor to cast in "int32" data type
tf.cast(x, 'int32').print();

Producción:

 Tensor
   [2, 1, 5, 0, 1, 0]

Ejemplo 2:

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using a tensor of some values
// as the parameter for .cast() function to
// cast into bool data type
tf.cast(tf.tensor1d([0, 1, -3]), 'bool').print();

Producción:

Tensor
   [false, true, true]

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

Publicación traducida automáticamente

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