Function torch::nn::functional::max_pool2d_with_indices

• Defined in File pooling.h

Function Documentation

inline std::tuple<Tensor, Tensor> torch::nn::functional::max_pool2d_with_indices(const Tensor &input, const MaxPool2dFuncOptions &options)

See the documentation for torch::nn::functional::MaxPool2dFuncOptions class to learn what optional arguments are supported for this functional.

Example:

namespace F = torch::nn::functional;
F::max_pool2d_with_indices(x, F::MaxPool2dFuncOptions(3).stride(2));