conv2d 在pytorch ， tf 中数据格式

conv2d 在pytorch ，tf中数据格式

pytorch data type is "NCHW"
but tensorflow data type is "NHWC",
when pytorch to onnx, and onnx to tensorflow how this diff process?

onnx only has NCHW support.
Thus, onnx-tf will export a NCHW based graph if you use TensorflowRep.export_graph method.

Conv2D	onnx	pytorch	tf
data format	NCHW	NCHW	NHWC