Tensorflow、CNN：GPUを使用しているときに 'NCHW'形式の畳み込みニューラルネットワークを作成する必要がありますか？

Question

DQNの実装コード（https://github.com/devsisters/DQN-tensorflow）を確認したところ、このコードではNCHW形式のCNN形式のネットワークが生成されていました。GPU,NHWCなどを使用すると、このコードが生成されます。

私はGPUを使用していてもNHWC形式のCNNネットワークを生成してもいいですか？

Answer 1

パフォーマンスについて気にしない場合は、大丈夫かもしれません。一般に、GPUではNCHWが高速です。

GPUでNHWCを特に使用したいのはなぜですか？ これらの間で変換することができます。 Convert between NHWC and NCHW in TensorFlow