Created by: zhiqiu
As the title, Pool2D supports NHWC to achieve better performance.
paddle.fludi.dygraph.Pool2D
add a keyword argument data_format
, which indicates the data format of the input and output data. An optional string from: "NCHW"
, "NHWC"
. The default is "NCHW"