Función torch.ravel de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.ravelEs una función de PyTorch que se utiliza para aplanar un tensor a una dimensión. Devuelve una vista del tensor de entrada (si es posible), no una copia.

Definición de la función

torch.ravel(input)

Descripción de parámetros:

  • input: tensor de entrada

Ejemplo de uso

Ejemplo

import torch

# Crear una matriz de 2x3
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Aplanar a una dimensión
y = torch.ravel(x)
print(y)

El resultado de salida es:

tensor([1, 2, 3, 4, 5, 6])

Ejemplo

import torch

# Crear un tensor tridimensional
x = torch.arange(12).reshape(2, 3, 2)

# Aplanar a una dimensión
y = torch.ravel(x)
print(y)

El resultado de salida es:

tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones