有時候在跑代碼是會出現(xiàn)下面這個錯誤
/root/miniconda3/lib/python3.8/site-packages/torch/functional.py:568: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2228.)

這個UserWarning警告是關于torch.meshgrid函數(shù)的改動提示。
在PyTorch的新版本中，torch.meshgrid函數(shù)需要增加一個indexing參數(shù)來指定索引模式。
目前torch.meshgrid默認的索引模式是'ij'，新版本中需要顯式指定:

根據(jù)報錯的提示點擊上述報錯提示內(nèi)容，然后跳轉(zhuǎn)到相應的functional[.py文件]

    kwargs = {} if indexing is None else {'indexing': indexing}
    return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]

改為

    # 新版本中需要指定indexing模式
    kwargs = {} if indexing is None else {'indexing': indexing}
    return _VF.meshgrid(tensors, **kwargs,  indexing='ij')  # type: ignore[attr-defined]

x = torch.arange(4)
y = torch.arange(3) 

xx, yy = torch.meshgrid(x, y) # 默認'ij'模式

# 新版本中需要指定indexing模式
xx, yy = torch.meshgrid(x, y, indexing='ij')

這個indexing參數(shù)有兩種模式:

'ij'模式: 返回的第一個矩陣索引是i，第二個矩陣索引是j，這是默認的傳統(tǒng)矩陣索引法則。
'xy'模式: 返回的第一個矩陣索引是x，第二個矩陣索引是y。
所以為了兼容新版本PyTorch，使用torch.meshgrid時需要顯式添加indexing='ij'參數(shù)。

這個UserWarning提醒我們新版本的改動，需要注意傳入索引模式，以避免默認模式改變后代碼出錯。