Fonction torch.tile de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.tileC'est une fonction de PyTorch utilisée pour répéter un tenseur plusieurs fois.

Définition de fonction

torch.tile(input, dims)

Exemples d'utilisation

Exemple

import torch

# Répétition de tenseur 1D
x = torch.tensor([1, 2, 3])
print("Tenseur 1D original :")
print(x)

result = torch.tile(x, 3)
print("Répété 3 fois :")
print(result)

# Répétition de tenseur 2D
y = torch.tensor([[1, 2], [3, 4]])
print("\nTenseur 2D original :")
print(y)

result = torch.tile(y, (2, 3))
print("Répété 2 fois le long des lignes, 3 fois le long des colonnes :")
print(result)

# Répétition de tenseur 3D
z = torch.tensor([[[1, 2], [3, 4]]])
print("\nTenseur 3D d'origine :")
print(z)

result = torch.tile(z, (2, 1, 2))
print("Forme [2, 1, 2] répétée :")
print(result)
print(f"Forme : {result.shape}")

La sortie est :

原始一维张量:
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])

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions