torch_geometric.datasets.graph_generator.GridGraph

class GridGraph(height: int, width: int, dtype: Optional[dtype] = None)[source]

Bases: GraphGenerator

Generates two-dimensional grid graphs. See grid() for more information.

Parameters:

  • height (int) – The height of the grid.

  • width (int) – The width of the grid.

  • dtype (torch.dtype, optional) – The desired data type of the returned position tensor. (default: None)