Class AvgPoolGrad.Options

java.lang.Object
org.tensorflow.op.nn.AvgPoolGrad.Options
Enclosing class:
AvgPoolGrad<T extends TNumber>

public static class AvgPoolGrad.Options extends Object
Optional attributes for AvgPoolGrad
  • Method Details

    • dataFormat

      public AvgPoolGrad.Options dataFormat(String dataFormat)
      Sets the dataFormat option.
      Parameters:
      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].
      Returns:
      this Options instance.