tf.raw_ops.MaxPoolGradWithArgmax
Computes gradients of the maxpooling function.
tf.raw_ops.MaxPoolGradWithArgmax( input, grad, argmax, ksize, strides, padding, include_batch_in_index=False, name=None )
Args | |
---|---|
input | A Tensor . Must be one of the following types: float32 , float64 , int32 , uint8 , int16 , int8 , int64 , bfloat16 , uint16 , half , uint32 , uint64 . The original input. |
grad | A Tensor . Must have the same type as input . 4-D with shape [batch, height, width, channels] . Gradients w.r.t. the output of max_pool . |
argmax | A Tensor . Must be one of the following types: int32 , int64 . The indices of the maximum values chosen for each output of max_pool . |
ksize | A list of ints that has length >= 4 . The size of the window for each dimension of the input tensor. |
strides | A list of ints that has length >= 4 . The stride of the sliding window for each dimension of the input tensor. |
padding | A string from: "SAME", "VALID" . The type of padding algorithm to use. |
include_batch_in_index | An optional bool . Defaults to False . Whether to include batch dimension in flattened index of argmax . |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as input . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/MaxPoolGradWithArgmax