torch.meshgrid()

x1、y1 = torch.meshgrid(x、y)
パラメータは2つであり、最初のパラメータはxであると想定し、2番目のパラメータはyが
2つのテンソルを出力すると想定し、サイズはx.size * y.size(行の数はxの数であり、列の数はyの数です)
具体的な出力を以下に示します。

注:2つのパラメーターのデータ型は、floatまたはintのどちらかである必要があります。そうでない場合、エラーが報告されます。
ここに画像の説明を挿入

943の元の記事を公開 136のような 330,000以上を訪問

おすすめ

転載: blog.csdn.net/weixin_36670529/article/details/105350139