Функция torch.column_stack в PyTorch
Справочное руководство Pytorch torch
torch.column_stackявляется функцией в PyTorch, используемой для объединения тензоров по столбцам.
Определение функции
torch.column_stack(tensors, *, out=None)
Пример использования
Пример
import torch
# Объединение 1D тензоров по столбцам
x1 = torch.tensor([1, 2, 3])
x2 = torch.tensor([4, 5, 6])
result = torch.column_stack([x1, x2])
print("Объединение 1D тензоров по столбцам:")
print(f" x1: {x1}")
print(f" x2: {x2}")
print(f" column_stack:n{result}")
# Объединение 2D тензоров по столбцам
y1 = torch.tensor([[1], [2], [3]])
y2 = torch.tensor([[4], [5], [6]])
result = torch.column_stack([y1, y2])
print("nОбъединение 2D тензоров по столбцам:")
print(f" y1:n{y1}")
print(f" y2:n{y2}")
print(f" column_stack:n{result}")
# Объединение нескольких тензоров по столбцам
z1 = torch.tensor([1, 2, 3])
z2 = torch.tensor([4, 5, 6])
z3 = torch.tensor([7, 8, 9])
result = torch.column_stack([z1, z2, z3])
print("nОбъединение нескольких 1D тензоров по столбцам:")
print(f" result:n{result}")
# Объединение 1D тензоров по столбцам
x1 = torch.tensor([1, 2, 3])
x2 = torch.tensor([4, 5, 6])
result = torch.column_stack([x1, x2])
print("Объединение 1D тензоров по столбцам:")
print(f" x1: {x1}")
print(f" x2: {x2}")
print(f" column_stack:n{result}")
# Объединение 2D тензоров по столбцам
y1 = torch.tensor([[1], [2], [3]])
y2 = torch.tensor([[4], [5], [6]])
result = torch.column_stack([y1, y2])
print("nОбъединение 2D тензоров по столбцам:")
print(f" y1:n{y1}")
print(f" y2:n{y2}")
print(f" column_stack:n{result}")
# Объединение нескольких тензоров по столбцам
z1 = torch.tensor([1, 2, 3])
z2 = torch.tensor([4, 5, 6])
z3 = torch.tensor([7, 8, 9])
result = torch.column_stack([z1, z2, z3])
print("nОбъединение нескольких 1D тензоров по столбцам:")
print(f" result:n{result}")
Результат вывода:
一维张量列堆叠:
x1: tensor([1, 2, 3])
x2: tensor([4, 5, 6])
column_stack:
tensor([[1, 4],
[2, 5],
[3, 6]])
二维张量列堆叠:
y1:
tensor([[1],
[2],
[3]])
y2:
tensor([[4],
[5],
[6]])
column_stack:
tensor([[1, 4],
[2, 5],
[3, 6]])
多个一维张量列堆叠:
result:
tensor([[1, 4, 7],
[2, 5, 8],
[3, 6, 9]])
Другие расширения