Función torch.atleast_2d de PyTorch

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

torch.atleast_2dEs una función en PyTorch que se utiliza para convertir tensores de entrada a al menos 2 dimensiones. Si la dimensión de entrada tiene menos de 2 dimensiones, se añaden dimensiones automáticamente para convertirlo en 2 dimensiones.

Definición de la función

torch.atleast_2d(*tensors)

Ejemplo de uso

]) Forma: torch.Size([1, 1])

import torch

# Convertir escalar a tensor 2D
x = torch.atleast_2d(5)
print("Escalar convertido:", x, "Forma:", x.shape)
# Salida: Escalar convertido: tensor([

# Convertir tensor 1D a 2D
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("De 1D a 2D:", y, "Forma:", y.shape)
# Salida: De 1D a 2D: tensor([[1, 2, 3]]) Forma: torch.Size([1, 3])

# El tensor que ya es 2D se mantiene sin cambios
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("Tensor 2D:", y, "Forma:", y.shape)

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

Otras extensiones