Modifier and Type | Class and Description |
---|---|
static class |
MaxPoolGradGradWithArgmax.Options
Optional attributes for
MaxPoolGradGradWithArgmax |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number,U extends Number> |
create(Scope scope,
Operand<T> input,
Operand<T> grad,
Operand<U> argmax,
List<Long> ksize,
List<Long> strides,
String padding,
MaxPoolGradGradWithArgmax.Options... options)
Factory method to create a class wrapping a new MaxPoolGradGradWithArgmax operation.
|
static MaxPoolGradGradWithArgmax.Options |
includeBatchInIndex(Boolean includeBatchInIndex) |
Output<T> |
output()
Gradients of gradients w.r.t.
|
equals, hashCode, op, toString
public static <T extends Number,U extends Number> MaxPoolGradGradWithArgmax<T> create(Scope scope, Operand<T> input, Operand<T> grad, Operand<U> argmax, List<Long> ksize, List<Long> strides, String padding, MaxPoolGradGradWithArgmax.Options... options)
scope
- current scopeinput
- The original input.grad
- 4-D with shape `[batch, height, width, channels]`. Gradients w.r.t. the
input of `max_pool`.argmax
- The indices of the maximum values chosen for each output of `max_pool`.ksize
- The size of the window for each dimension of the input tensor.strides
- The stride of the sliding window for each dimension of the
input tensor.padding
- The type of padding algorithm to use.options
- carries optional attributes valuespublic static MaxPoolGradGradWithArgmax.Options includeBatchInIndex(Boolean includeBatchInIndex)
includeBatchInIndex
- Whether to include batch dimension in flattened index of `argmax`.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 extends Number>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.