Fonction torch.flipud de PyTorch

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

torch.flipudUne fonction de PyTorch utilisée pour retourner un tenseur de haut en bas (inversion verticale).

Définition de fonction

torch.flipud(input)

Exemples d'utilisation

Exemple

import torch

# Retourner un tenseur 2D de haut en bas
x = torch.arange(12).reshape(3, 4)
print("Tenseur d'origine :")
print(x)

result = torch.flipud(x)
print("Inversé de haut en bas :")
print(result)

# Inverser une matrice carrée de haut en bas
y = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("matrice carrée 3x3 :")
print(y)

result = torch.flipud(y)
print("Inversé de haut en bas :")
print(result)

# Inverser un tenseur 1D de haut en bas
z = torch.tensor([1, 2, 3, 4])
result = torch.flipud(z)
print("tenseur 1D [1, 2, 3, 4] :")
print("Inversé de haut en bas :", result)

La sortie est :

原始张量:
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
上下翻转:
tensor([[ 8,  9, 10, 11],
        [ 4,  5,  6,  7],
        [ 0,  1,  2,  3]])

3x3 方阵:
tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])
上下翻转:
tensor([[7, 8, 9],
        [4, 5, 6],
        [1, 2, 3]])

一维张量 [1, 2, 3, 4]:
上下翻转: tensor([4, 3, 2, 1])

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

Autres extensions