Fonction torch.meshgrid de PyTorch

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

torch.meshgridC'est une fonction dans PyTorch utilisée pour créer une grille à partir de vecteurs de coordonnées. Elle génère des tenseurs de coordonnées à partir de plusieurs vecteurs de coordonnées unidimensionnels.

Définition de fonction

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

Description des paramètres :

  • *tensors: Tenseur unidimensionnel d'entrée
  • indexing: Mode d'indexation, 'xy' (cartésien) ou 'ij' (matrice)

Exemple d'utilisation

Exemple

import torch

# Créer des vecteurs de coordonnées
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Créer un meshgrid (indexation cartésienne)
grid_x, grid_y = torch.meshgrid(x, y, indexing='xy')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

La sortie est :

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

Exemple

import torch

# Créer des vecteurs de coordonnées
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])

# Créer un meshgrid (indexation matricielle)
grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)

La sortie est :

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

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

Autres extensions