Conv2d.Options

public static class Conv2d.Options

Optional attributes for Conv2d

Public Methods

Conv2d.Options
dataFormat (String dataFormat)
Conv2d.Options
dilations (List<Long> dilations)
Conv2d.Options
explicitPaddings (List<Long> explicitPaddings)
Conv2d.Options
useCudnnOnGpu (Boolean useCudnnOnGpu)

Inherited Methods

Public Methods

public Conv2d.Options dataFormat (String dataFormat)

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, height, width, channels]. Alternatively, the format could be "NCHW", the data storage order of: [batch, channels, height, width].

public Conv2d.Options dilations (List<Long> dilations)

Parameters
dilations 1-D tensor of length 4. The dilation factor for each dimension of `input`. If set to k > 1, there will be k-1 skipped cells between each filter element on that dimension. The dimension order is determined by the value of `data_format`, see above for details. Dilations in the batch and depth dimensions must be 1.

public Conv2d.Options explicitPaddings (List<Long> explicitPaddings)

Parameters
explicitPaddings If `padding` is `"EXPLICIT"`, the list of explicit padding amounts. For the ith dimension, the amount of padding inserted before and after the dimension is `explicit_paddings[2 * i]` and `explicit_paddings[2 * i + 1]`, respectively. If `padding` is not `"EXPLICIT"`, `explicit_paddings` must be empty.

public Conv2d.Options useCudnnOnGpu (Boolean useCudnnOnGpu)