“¿Cómo convierto un tensor en una matriz numpy?” Código de respuesta

¿Cómo convierto un tensor en una matriz numpy?

import torch

# Create PyTorch tensor
A_torch = torch.tensor([1, 2])

# Convert tensor to NumPy array
A_np = A_torch.numpy()
AI Nerd

convertir el tensor en una matriz numpy

import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])                 
b = tf.add(a, 1)

a.numpy()
# array([[1, 2],
#        [3, 4]], dtype=int32)

b.numpy()
# array([[2, 3],
#        [4, 5]], dtype=int32)

tf.multiply(a, b).numpy()
# array([[ 2,  6],
#        [12, 20]], dtype=int32)
Shanti

Respuestas similares a “¿Cómo convierto un tensor en una matriz numpy?”

Preguntas similares a “¿Cómo convierto un tensor en una matriz numpy?”

Más respuestas relacionadas con “¿Cómo convierto un tensor en una matriz numpy?” en Python

Explore las respuestas de código populares por idioma

Explorar otros lenguajes de código