PyTorch torch.meshgrid-Funktion
PyTorch torch Referenzhandbuch
torch.meshgridEs ist eine Funktion in PyTorch, die verwendet wird, um ein Gitter aus Koordinatenvektoren zu erstellen. Sie erzeugt Koordinatentensoren aus mehreren eindimensionalen Koordinatenvektoren.
Funktionsdefinition
torch.meshgrid(*tensors, indexing='xy')
Parameterbeschreibung:
*tensors: Eindimensionaler Eingabetensorindexing: Indizierungsmodus, 'xy' (kartesisch) oder 'ij' (Matrix)
Beispielverwendung
Beispiel
import torch
# Koordinatenvektoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])
# Meshgrid erstellen (kartesische Indizierung)
grid_x, grid_y = torch.meshgrid(x, y, indexing='xy')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)
# Koordinatenvektoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])
# Meshgrid erstellen (kartesische Indizierung)
grid_x, grid_y = torch.meshgrid(x, y, indexing='xy')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)
Die Ausgabe ist:
grid_x:
tensor([[1, 2, 3],
[1, 2, 3]])
grid_y:
tensor([[4, 4, 4],
[5, 5, 5]])
Beispiel
import torch
# Koordinatenvektoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])
# Meshgrid erstellen (Matrixindizierung)
grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)
# Koordinatenvektoren erstellen
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5])
# Meshgrid erstellen (Matrixindizierung)
grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
print("grid_x:")
print(grid_x)
print("grid_y:")
print(grid_y)
Die Ausgabe ist:
grid_x:
tensor([[1, 1],
[2, 2],
[3, 3]])
grid_y:
tensor([[4, 5],
[4, 5],
[4, 5]])
Andere Erweiterungen