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

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

torch.meshgridЭто функция в PyTorch, используемая для создания сетки из координатных векторов. Она генерирует координатные тензоры из нескольких одномерных координатных векторов.

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

torch.meshgrid(*tensors, indexing='xy')

Описание параметров:

  • *tensors: Входной одномерный тензор
  • indexing: Режим индексации, 'xy' (декартова) или 'ij' (матричная)

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

Пример

import torch

# Создать координатные векторы
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Создать meshgrid (декартова индексация)
grid_x, grid_y = torch.meshgrid(x, y, indexing='xy')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

Вывод:

grid_x:
tensor([[1, 2, 3],
        [1, 2, 3]])
grid_y:
tensor([[4, 4, 4],
        [5, 5, 5]])

Пример

import torch

# Создать координатные векторы
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Создать meshgrid (матричная индексация)
grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

Вывод:

grid_x:
tensor([[1, 1],
        [2, 2],
        [3, 3]])
grid_y:
tensor([[4, 5],
        [4, 5],
        [4, 5]])

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

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