max_pool2d¶
dragon.vm.torch.nn.functional.max_pool2d(
input,
kernel_size,
stride=1,
padding=0,
ceil_mode=False
)[source]¶Apply the 2d max pooling to input.
- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- kernel_size (Union[int, Sequence[int]]) – The size of pooling window.
- stride (Union[int, Sequence[int]], optional, default=1) – The stride of pooling window.
- padding (Union[int, Sequence[int]], optional, default=0) – The zero padding size.
- ceil_mode (bool, optional, default=False) – Ceil or floor the boundary.
- Returns:
dragon.vm.torch.Tensor – The output tensor.
See also