Функция torch.atleast_2d PyTorch
Справочное руководство PyTorch torch
torch.atleast_2d— функция в PyTorch, используемая для преобразования входных тензоров как минимум к 2 измерениям. Если вход имеет менее 2 измерений, она автоматически добавляет измерения, чтобы сделать его 2-мерным.
Определение функции
torch.atleast_2d(*tensors)
Пример использования
]) Форма: torch.Size([1, 1])
import torch
# Преобразовать скаляр в 2-мерный тензор
x = torch.atleast_2d(5)
print("После преобразования скаляра:", x, "Форма:", x.shape)
# Вывод: После преобразования скаляра: tensor([
# Преобразовать 1-мерный тензор в 2 измерения
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1D to 2D:", y, "Shape:", y.shape)
# Вывод: 1D to 2D: tensor([[1, 2, 3]]) Shape: torch.Size([1, 3])
# Тензор, который уже является 2-мерным, остаётся без изменений
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2D tensor:", y, "Shape:", y.shape)
# Преобразовать скаляр в 2-мерный тензор
x = torch.atleast_2d(5)
print("После преобразования скаляра:", x, "Форма:", x.shape)
# Вывод: После преобразования скаляра: tensor([
# Преобразовать 1-мерный тензор в 2 измерения
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1D to 2D:", y, "Shape:", y.shape)
# Вывод: 1D to 2D: tensor([[1, 2, 3]]) Shape: torch.Size([1, 3])
# Тензор, который уже является 2-мерным, остаётся без изменений
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2D tensor:", y, "Shape:", y.shape)
Другие расширения