org.tensorflow.op.nn

Class AvgPool3d.Options

java.lang.Object

org.tensorflow.op.nn.AvgPool3d.Options

Enclosing class:

AvgPool3d<T extends Number>

public static class AvgPool3d.Options
extends Object

Optional attributes for AvgPool3d

Method Summary

Modifier and Type	Method and Description
AvgPool3d.Options	dataFormat(String dataFormat)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

dataFormat

public AvgPool3d.Options dataFormat(String dataFormat)

Parameters:

dataFormat - The data format of the input and output data. With the default format "NDHWC", the data is stored in the order of: [batch, in_depth, in_height, in_width, in_channels]. Alternatively, the format could be "NCDHW", the data storage order is: [batch, in_channels, in_depth, in_height, in_width].