T
- data type for output()
output@Operator(group="nn") public final class BiasAddGrad<T> extends PrimitiveOp implements Operand<T>
It accumulates all the values from out_backprop into the feature dimension. For NHWC data format, the feature dimension is the last. For NCHW data format, the feature dimension is the third-to-last.
Modifier and Type | Class and Description |
---|---|
static class |
BiasAddGrad.Options
Optional attributes for
BiasAddGrad |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> BiasAddGrad<T> |
create(Scope scope,
Operand<T> outBackprop,
BiasAddGrad.Options... options)
Factory method to create a class wrapping a new BiasAddGrad operation.
|
static BiasAddGrad.Options |
dataFormat(String dataFormat) |
Output<T> |
output()
1-D with size the feature dimension of `out_backprop`.
|
equals, hashCode, op, toString
public static <T> BiasAddGrad<T> create(Scope scope, Operand<T> outBackprop, BiasAddGrad.Options... options)
scope
- current scopeoutBackprop
- Any number of dimensions.options
- carries optional attributes valuespublic static 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.public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.