public static class
BiasAdd.Options
Optional attributes for BiasAdd
Public Methods
BiasAdd.Options |
dataFormat(String dataFormat)
|
Inherited Methods
Public Methods
public BiasAdd.Options dataFormat (String dataFormat)
Parameters
dataFormat | Specify the data format of the input and output data. With the default format "NHWC", the bias tensor will be added to the last dimension of the value tensor. Alternatively, the format could be "NCHW", the data storage order of: [batch, in_channels, in_height, in_width]. The tensor will be added to "in_channels", the third-to-the-last dimension. |
---|