Fonction torch.flip de PyTorch

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

torch.flipIl s'agit d'une fonction de PyTorch utilisée pour inverser un tenseur selon les dimensions spécifiées.

Définition de fonction

torch.flip(input, dims)

Exemple d'utilisation

Exemple

import torch

x = torch.arange(8).reshape(2, 4)

print("Original:")
print(x)

print("Inversion horizontale :")
print(torch.flip(x, dims=[1]))

print("Inversion verticale :")
print(torch.flip(x, dims=[0]))

Le résultat de sortie est:

原始:
tensor([[0, 1, 2, 3],
        [4, 5, 6, 7]])
水平翻转:
tensor([[3, 2, 1, 0],
        [7, 6, 5, 4]])
垂直翻转:
tensor([[4, 5, 6, 7],
        [0, 1, 2, 3]])

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

Autres extensions