,

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.atleast_3d— это функция в PyTorch, используемая для преобразования входных тензоров как минимум в 3 измерения. Если входные данные имеют менее 3 измерений, измерения автоматически добавляются, чтобы сделать их 3-мерными.

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

torch.atleast_3d(*tensors)

Пример использования

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

import torch

# Преобразовать скаляр в 3D-тензор
x = torch.atleast_3d(5)
print("После преобразования скаляра:", x, "Форма:", x.shape)
# Вывод: После преобразования скаляра: tensor([[

# Преобразовать 1D-тензор в 3D
x = torch.tensor([1, 2, 3])
y = torch.atleast_3d(x)
print("1D в 3D:", y, "Форма:", y.shape)
# Вывод: 1D в 3D: tensor([[

# Преобразовать 2D-тензор в 3D
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_3d(x)
print("2D в 3D:", y, "Форма:", y.shape)
# Вывод: 2D в 3D: tensor([[[1, 2], [3, 4]]]) Форма: torch.Size([1, 2, 2])

# Тензор, который уже является 3D, остаётся неизменным
x = torch.tensor([[[1, 2], [3, 4]]])
y = torch.atleast_3d(x)
print("3D-тензор:", y, "Форма:", y.shape)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения