nnf_adaptive_max_pool2d: Adaptive_max_pool2d

View source: R/nnf-pooling.R

nnf_adaptive_max_pool2dR Documentation

Adaptive_max_pool2d

Description

Applies a 2D adaptive max pooling over an input signal composed of several input planes.

Usage

nnf_adaptive_max_pool2d(input, output_size, return_indices = FALSE)

Arguments

input

input tensor (minibatch, in_channels , iH , iW)

output_size

the target output size (single integer or double-integer tuple)

return_indices

whether to return pooling indices. Default: FALSE


torch documentation built on Oct. 24, 2022, 5:08 p.m.