nnf_max_unpool2d function

Max_unpool2d

Computes a partial inverse of MaxPool2d.

nnf_max_unpool2d(
  input,
  indices,
  kernel_size,
  stride = NULL,
  padding = 0,
  output_size = NULL
)

Arguments

• input: the input Tensor to invert
• indices: the indices given out by max pool
• kernel_size: Size of the max pooling window.
• stride: Stride of the max pooling window. It is set to kernel_size by default.
• padding: Padding that was added to the input
• output_size: the targeted output size