Функция torch.tile в PyTorch
Справочное руководство Pytorch torch
torch.tileЭто функция в PyTorch, используемая для повторения тензора несколько раз.
Определение функции
torch.tile(input, dims)
Примеры использования
Пример
import torch
# Повторение 1D-тензора
x = torch.tensor([1, 2, 3])
print("Исходный 1D-тензор:")
print(x)
result = torch.tile(x, 3)
print("Повторено 3 раза:")
print(result)
# Повторение 2D-тензора
y = torch.tensor([[1, 2], [3, 4]])
print("\nИсходный 2D-тензор:")
print(y)
result = torch.tile(y, (2, 3))
print("Повторено 2 раза по строкам, 3 раза по столбцам:")
print(result)
# Повторение 3D-тензора
z = torch.tensor([[[1, 2], [3, 4]]])
print("\nИсходный 3D-тензор:")
print(z)
result = torch.tile(z, (2, 1, 2))
print("Форма [2, 1, 2] повторена:")
print(result)
print(f"Форма: {result.shape}")
# Повторение 1D-тензора
x = torch.tensor([1, 2, 3])
print("Исходный 1D-тензор:")
print(x)
result = torch.tile(x, 3)
print("Повторено 3 раза:")
print(result)
# Повторение 2D-тензора
y = torch.tensor([[1, 2], [3, 4]])
print("\nИсходный 2D-тензор:")
print(y)
result = torch.tile(y, (2, 3))
print("Повторено 2 раза по строкам, 3 раза по столбцам:")
print(result)
# Повторение 3D-тензора
z = torch.tensor([[[1, 2], [3, 4]]])
print("\nИсходный 3D-тензор:")
print(z)
result = torch.tile(z, (2, 1, 2))
print("Форма [2, 1, 2] повторена:")
print(result)
print(f"Форма: {result.shape}")
Вывод:
原始一维张量:
tensor([1, 2, 3])
重复 3 次:
tensor([1, 2, 3, 1, 2, 3, 1, 2, 3])
原始二维张量:
tensor([[1, 2],
[3, 4]])
沿行重复 2 次，沿列重复 3 次:
tensor([[1, 2, 1, 2, 1, 2],
[3, 4, 3, 4, 3, 4],
[1, 2, 1, 2, 1, 2],
[3, 4, 3, 4, 3, 4]])
原始三维张量:
tensor([[[1, 2],
[3, 4]]])
形状 [2, 1, 2] 重复:
tensor([[[1, 2, 1, 2],
[3, 4, 3, 4]],
[[1, 2, 1, 2],
[3, 4, 3, 4]]])
形状: torch.Size([2, 2, 4])
Другие расширения