public static class AvgPoolGrad.Options extends Object
AvgPoolGrad
Modifier and Type | Method and Description |
---|---|
AvgPoolGrad.Options |
dataFormat(String dataFormat) |
public AvgPoolGrad.Options dataFormat(String dataFormat)
dataFormat
- Specify the data format of the input and output data. With the
default format "NHWC", the data is stored in the order of:
[batch, in_height, in_width, in_channels].
Alternatively, the format could be "NCHW", the data storage order of:
[batch, in_channels, in_height, in_width].Copyright © 2022. All rights reserved.