PyTorch torch.tile 関数
Pytorch torch リファレンスマニュアルtorch.tileは PyTorch においてテンソルを複数回繰り返すために使用される関数です。
関数定義
torch.tile(input, dims)
使用例
実例
import torch
# 1次元テンソルの繰り返し
x = torch.tensor([1, 2, 3])
print("元の1次元テンソル:")
print(x)
result = torch.tile(x, 3)
print("3回繰り返し:")
print(result)
# 2次元テンソルの繰り返し
y = torch.tensor([[1, 2], [3, 4]])
print("n元の2次元テンソル:")
print(y)
result = torch.tile(y, (2, 3))
print("行方向に2回、列方向に3回繰り返し:")
print(result)
# 3次元テンソルの繰り返し
z = torch.tensor([[[1, 2], [3, 4]]])
print("n元の3次元テンソル:")
print(z)
result = torch.tile(z, (2, 1, 2))
print("形状 [2, 1, 2] 繰り返し:")
print(result)
print(f"形状: {result.shape}")
# 1次元テンソルの繰り返し
x = torch.tensor([1, 2, 3])
print("元の1次元テンソル:")
print(x)
result = torch.tile(x, 3)
print("3回繰り返し:")
print(result)
# 2次元テンソルの繰り返し
y = torch.tensor([[1, 2], [3, 4]])
print("n元の2次元テンソル:")
print(y)
result = torch.tile(y, (2, 3))
print("行方向に2回、列方向に3回繰り返し:")
print(result)
# 3次元テンソルの繰り返し
z = torch.tensor([[[1, 2], [3, 4]]])
print("n元の3次元テンソル:")
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])
その他の拡張