Функция torch.flipud в PyTorch


Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.flipudФункция в PyTorch, используемая для переворота тензора сверху вниз (вертикального отражения).

Определение функции

torch.flipud(input)

Примеры использования

Пример

import torch

# Отразить 2D-тензор сверху вниз
x = torch.arange(12).reshape(3, 4)
print("Исходный тензор:")
print(x)

result = torch.flipud(x)
print("Перевёрнуто вверх дном:")
print(result)

# Перевернуть квадратную матрицу вверх дном
y = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("Квадратная матрица 3x3:")
print(y)

result = torch.flipud(y)
print("Перевёрнуто вверх дном:")
print(result)

# Перевернуть 1D-тензор вверх дном
z = torch.tensor([1, 2, 3, 4])
result = torch.flipud(z)
print("1D-тензор [1, 2, 3, 4]:")
print("Перевёрнуто вверх дном:", result)

Вывод:

原始张量:
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 参考手册Справочное руководство PyTorch torch

Другие расширения