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.
|
