Class MaxPool1dImpl¶
Defined in File pooling.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::MaxPoolImpl< 1, MaxPool1dImpl >(Template Class MaxPoolImpl)
Class Documentation¶
-
class MaxPool1dImpl : public torch::nn::MaxPoolImpl<1, MaxPool1dImpl>¶
Applies maxpool over a 1-D input.
See https://pytorch.org/docs/main/nn.html#torch.nn.MaxPool1d to learn about the exact behavior of this module.
See the documentation for
torch::nn::MaxPool1dOptionsclass to learn what constructor arguments are supported for this module.Example:
MaxPool1d model(MaxPool1dOptions(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::MaxUnpool1dlater.
-
Tensor forward(const Tensor &input)¶
