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