,

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

torch.atleast_3dEs una función de PyTorch que se utiliza para convertir el tensor de entrada a al menos 3 dimensiones. Si la entrada tiene menos de 3 dimensiones, se agregan dimensiones automáticamente para que sea tridimensional.

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

torch.atleast_3d(*tensors)

Ejemplo de uso

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

import torch

# Convertir un escalar a tensor tridimensional
x = torch.atleast_3d(5)
print("Después de convertir el escalar:", x, "Forma:", x.shape)
# Salida: Después de convertir el escalar: tensor([[

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

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

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

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

Otras extensiones