Función torch.atleast_1d de PyTorch

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

torch.atleast_1dEs una función en PyTorch utilizada para convertir el tensor de entrada a al menos 1 dimensión. Si la entrada ya tiene al menos 1 dimensión, devuelve una vista del tensor original; de lo contrario, añade una nueva dimensión.

Definición de la función

torch.atleast_1d(*tensors)

Ejemplo de uso

Ejemplo

import torch

# Convertir escalar a tensor de 1 dimensión
x = torch.atleast_1d(5)
print("Después de la conversión del escalar:", x, "Forma:", x.shape)

# Tensor que ya es de 1 dimensión
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("Tensor 1D:", y, "Forma:", y.shape)

# El tensor de 2 dimensiones permanece igual
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("Tensor 2D:", y, "Forma:", y.shape)

# Múltiples entradas
a = 1
b = torch.tensor([2, 3])
c = torch.tensor([[4, 5]])
result = torch.atleast_1d(a, b, c)
for i, t in enumerate(result):
    print(f"Forma de entrada {i}: {t.shape}")

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

Otras extensiones