max_pool1d

dragon.vm.torch.nn.functional.max_pool1d(
  input,
  kernel_size,
  stride=1,
  padding=0,
  ceil_mode=False
)[source]

Apply the 1d 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.