Функция torch.atleast_1d PyTorch
Справочное руководство PyTorch torch
torch.atleast_1d— это функция в PyTorch, используемая для преобразования входных тензоров к размерности не менее 1. Если вход уже имеет размерность не менее 1, она возвращает представление исходного тензора; в противном случае она добавляет новое измерение.
Определение функции
torch.atleast_1d(*tensors)
Пример использования
Пример
import torch
# Скаляр преобразован в 1-D тензор
x = torch.atleast_1d(5)
print("После преобразования скаляра:", x, "Форма:", x.shape)
# Тензор уже 1-D
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("1-D тензор:", y, "Форма:", y.shape)
# 2-D тензор остаётся без изменений
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("2-D тензор:", y, "Форма:", y.shape)
# Несколько входов
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"Форма входа {i}: {t.shape}")
# Скаляр преобразован в 1-D тензор
x = torch.atleast_1d(5)
print("После преобразования скаляра:", x, "Форма:", x.shape)
# Тензор уже 1-D
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("1-D тензор:", y, "Форма:", y.shape)
# 2-D тензор остаётся без изменений
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("2-D тензор:", y, "Форма:", y.shape)
# Несколько входов
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"Форма входа {i}: {t.shape}")
Другие расширения