まず、Tensorflowはチャネルの前後の画像入力チャネルの2つのモードをサポートしますが、Pytorchはチャネルの前の入力モードのみをサポートします。
Tensorflowは、畳み込み層の入力として、最初のチャネルと最後のチャネルの2つの形式のデータをサポートします。最初の形式では、データは[N、C、H、W]の形式になります。ここで、Nはバッチサイズ、Cはチャネル数、Hは行列の高さ、Wは行列の幅です。2番目の形式では、データは[N、H、W、C]の形式になります。PyTorchでは、チャネルの最初のデータ形式のみがサポートされています。
Tensorflowでは、2つのモードの切り替えと選択を設定できます。Pytorchでは、チャネルが遅れている場合、入力のみを変換できます。https：//blog.csdn.net/qq_40877238/article/を参照してください。詳細/ 106024442