3j=SSKJr SSKrS S\S\S\S\\R S\\R S\R4 S jjrS S \S\S\S\S\\R S\\R S\R4S jjr g) )OptionalNheightwidthnormalized_coordinatesdevicedtypereturncB[R"SUS- XUS9n[R"SUS- XUS9nU(aXQS- - S- S-nX`S- - S- S-n[R"[R"XV/SS9SS 9nUR SSS5R S5$) aGenerate a coordinate grid for an image. When the flag ``normalized_coordinates`` is set to True, the grid is normalized to be in the range :math:`[-1,1]` to be consistent with the pytorch function :py:func:`torch.nn.functional.grid_sample`. Args: height: the image height (rows). width: the image width (cols). normalized_coordinates: whether to normalize coordinates in the range :math:`[-1,1]` in order to be consistent with the PyTorch function :py:func:`torch.nn.functional.grid_sample`. device: the device on which the grid will be generated. dtype: the data type of the generated grid. Return: grid tensor with shape :math:`(1, H, W, 2)`. Example: >>> create_meshgrid(2, 2) tensor([[[[-1., -1.], [ 1., -1.]], [[-1., 1.], [ 1., 1.]]]]) >>> create_meshgrid(2, 2, normalized_coordinates=False) tensor([[[[0., 0.], [1., 0.]], [[0., 1.], [1., 1.]]]]) rrr?ijindexingdimtorchlinspacestackmeshgridpermute unsqueeze)rrrrrxsys base_grids N/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/geometry/grid.pycreate_meshgridr sR~~aEPUVB~~a!VRWXBAI$ )QJ#% *$kk%.."D*QWYZI   Q1 % / / 22depthc[R"SUS- X$US9n[R"SUS- XUS9n[R"SUS- XUS9nU(a'XbS- - S- S-nXqS- - S- S-nXS- - S- S-n[R"[R"XU/SS9SS 9n U R SSSS 5R S5$) aGenerate a coordinate grid for an image. When the flag ``normalized_coordinates`` is set to True, the grid is normalized to be in the range :math:`[-1,1]` to be consistent with the pytorch function :py:func:`torch.nn.functional.grid_sample`. Args: depth: the image depth (channels). height: the image height (rows). width: the image width (cols). normalized_coordinates: whether to normalize coordinates in the range :math:`[-1,1]` in order to be consistent with the PyTorch function :py:func:`torch.nn.functional.grid_sample`. device: the device on which the grid will be generated. dtype: the data type of the generated grid. Return: grid tensor with shape :math:`(1, D, H, W, 3)`. rr r r rrrrrr) r"rrrrrrrzsrs rcreate_meshgrid3dr&Xs8~~aEPUVB~~a!VRWXB~~aEPUVBAI$ )QJ#% *AI$ ) ENNBB<$GRPI   Q1a ( 2 21 55r!)TNN) typingrrintboolrrTensorr r&r!rr,s& $(%)#' =3 =3 =3!=3 U\\ " =3 EKK =3  \\ =3H$(%)#' &6 &6 &6 &6! &6 U\\ " &6 EKK &6 \\&6r!