在较新版本的 PyTorch 中,当使用 torch.meshgrid() 时如果没有指定 indexing 参数,会收到这个警告。这是为了确保与 NumPy 的 meshgrid 行为保持一致,并提高代码的明确性。

原来的版本是
 

x, y = torch.meshgrid(x, y)

现在需要加上

x, y = torch.meshgrid(tensor1, tensor2, indexing='ij')  # 保持原来行为
# 或
x, y = torch.meshgrid(tensor1, tensor2, indexing='xy')  # 使用笛卡尔坐标风格

不确定就选择第一种,亲身经历,使用第二种,代码跑出来没有效果。

这个变化是为了:

  1. 使 PyTorch 的行为与 NumPy 更加一致

  2. 让用户明确选择他们想要的网格坐标风格

  3. 避免在不同版本间出现意外行为变化

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐