Fonction torch.fliplr de PyTorch
Manuel de référence Pytorch torch
torch.fliplrEst une fonction de PyTorch utilisée pour inverser un tenseur de gauche à droite (inversion horizontale).
Définition de fonction
torch.fliplr(input)
Exemple d'utilisation
Exemple
import torch
# Retourner un tenseur 2D de gauche à droite
x = torch.arange(12).reshape(3, 4)
print("Tenseur d'origine :")
print(x)
result = torch.fliplr(x)
print("Inversion gauche-droite :")
print(result)
# Retourner une matrice carrée de gauche à droite
y = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("n3x3 matrice carrée :")
print(y)
result = torch.fliplr(y)
print("Inversion gauche-droite :")
print(result)
# Les vecteurs ne peuvent pas être retournés ; une erreur sera levée
z = torch.tensor([1, 2, 3])
try:
result = torch.fliplr(z)
except RuntimeError as e:
print(f"nErreur lors de l'inversion d'un vecteur 1D : {e}")
# Retourner un tenseur 2D de gauche à droite
x = torch.arange(12).reshape(3, 4)
print("Tenseur d'origine :")
print(x)
result = torch.fliplr(x)
print("Inversion gauche-droite :")
print(result)
# Retourner une matrice carrée de gauche à droite
y = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("n3x3 matrice carrée :")
print(y)
result = torch.fliplr(y)
print("Inversion gauche-droite :")
print(result)
# Les vecteurs ne peuvent pas être retournés ; une erreur sera levée
z = torch.tensor([1, 2, 3])
try:
result = torch.fliplr(z)
except RuntimeError as e:
print(f"nErreur lors de l'inversion d'un vecteur 1D : {e}")
Le résultat de sortie est:
原始张量:
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
左右翻转:
tensor([[ 3, 2, 1, 0],
[ 7, 6, 5, 4],
[11, 10, 9, 8]])
3x3 方阵:
tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
左右翻转:
tensor([[3, 2, 1],
[6, 5, 4],
[9, 8, 7]])
一维向量翻转报错: fliplr: Input must be at least 2-D