tensorflow.nn.conv2d与NCHW格式的滤波器形状

Question

以下Tensorflow's best practices for performance，我使用NCHW数据格式，但我不确定在tensorflow.nn.conv2d中使用的滤波器形状。

该文档说，使用NHWC格式的[filter_height, filter_width, in_channels, out_channels]，但不清楚如何处理NCHW。

应该使用相同的形状吗？

Answer 1

使用相同的过滤器形状应该工作。对函数参数的唯一改变是步幅。举个例子，假设您希望您的体系结构能够同时使用这两种格式，这也是推荐的：

# input -> Tensor in NCHW format 
if use_nchw: 
    result = tf.nn.conv2d(
     input=input, 
     filter=filter, 
     strides=[1, 1, stride, stride], 
     data_format='NCHW') 
else: 
    input_t = tf.transpose(input, [0, 2, 3, 1]) # NCHW to NHWC 

    result = tf.nn.conv2d(
     input=input_t, 
     filter=filter, 
     strides=[1, stride, stride, 1]) 

    result = tf.transpose(result, [0, 3, 1, 2]) # NHWC to NCHW