CPUへのテンソルフローの最適化

Question

CPUに最適化する必要があるモデルがあります。

現在、このモデルでは1024 x 1024バイトのデータが使用されています。このドキュメントを1として

images = img[y:y+1024,x:x+1024,:] 

、彼らはNHCWNCHWの形式から、デフォルトのtensorflowのデータ形式を変更したいです。

NHWCからNCHW形式に変換するにはどうすればよいですか？このドキュメントを1として

https://software.intel.com/en-us/articles/tensorflow-optimizations-on-modern-intel-architecture

Answer 1

、彼らはNCHW形式にNHCWからデフォルトtensorflowのデータ形式を変更したいです。

実際、NHCW形式をサポートするTensorflow関数は見たことがありません。たとえば、tf.nn.conv2dとtf.nn.conv2d_transposeは、NHWC（現在のデフォルト）とNCHWの形式をサポートしています。 tf.nn.max_poolはNHWC,NCHWおよびNCHW_VECT_Cをサポートしています（最後はcudnn6の量子化された畳み込みのための最も実績のあるテンソル形式です、NCHWに似ています）。

どのようにNHCW形式からNCHW形式に変換できますか？

しかし、この変換は可能です。同様に、高次元のテンソルで動作しますtf.transpose経由：

# NHCW 
original = tf.placeholder(dtype=tf.float32, shape=[None, 1024, 3, 1024]) 
# NCHW: swap 1 and 2 axis 
transformed = tf.transpose(original, perm=[0, 2, 1, 3]) 

またnp.swapaxes(array, 1, 2)経由numpyの中でこれを行うことができます。