org.tensorflow.op.nn

Class MaxPool3d.Options

java.lang.Object

org.tensorflow.op.nn.MaxPool3d.Options

Enclosing class:

MaxPool3d<T extends Number>

public static class MaxPool3d.Options
extends Object

Optional attributes for MaxPool3d

Method Summary

Modifier and Type	Method and Description
MaxPool3d.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 MaxPool3d.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].