Función torch.meshgrid de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.meshgridEs una función en PyTorch utilizada para crear una cuadrícula a partir de vectores de coordenadas. Genera tensores de coordenadas a partir de múltiples vectores de coordenadas unidimensionales.

Definición de la función

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

Descripción de parámetros:

  • *tensors: tensor unidimensional de entrada
  • indexing: modo de indexación, 'xy' (cartesiano) o 'ij' (matriz)

Ejemplo de uso

Ejemplo

import torch

# Crea vectores de coordenadas
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Crea la cuadrícula (indexación cartesiana)
grid_x, grid_y = torch.meshgrid(x, y, indexing='xy')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

El resultado de salida es:

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

Ejemplo

import torch

# Crea vectores de coordenadas
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Crea la cuadrícula (indexación de matriz)
grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones