public static class BiasAddGrad.Options extends Object
BiasAddGrad
Modifier and Type | Method and Description |
---|---|
BiasAddGrad.Options |
dataFormat(String dataFormat) |
public BiasAddGrad.Options dataFormat(String dataFormat)
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.Copyright © 2022. All rights reserved.