Class MaxPool3dImpl¶
Defined in File pooling.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::MaxPoolImpl< 3, MaxPool3dImpl >(Template Class MaxPoolImpl)
Class Documentation¶
-
class MaxPool3dImpl : public torch::nn::MaxPoolImpl<3, MaxPool3dImpl>¶
Applies maxpool over a 3-D input.
See https://pytorch.org/docs/main/nn.html#torch.nn.MaxPool3d to learn about the exact behavior of this module.
See the documentation for
torch::nn::MaxPool3dOptionsclass to learn what constructor arguments are supported for this module.Example:
MaxPool3d model(MaxPool3dOptions(3).stride(2));
Public Functions
-
Tensor forward(const Tensor &input)¶
-
std::tuple<Tensor, Tensor> forward_with_indices(const Tensor &input)¶
Returns the outputs and the indices of the max values.
Useful for
torch::nn::MaxUnpool3dlater.
-
Tensor forward(const Tensor &input)¶
