adaptive_avg_pool1d

dragon.vm.torch.nn.functional.adaptive_avg_pool1d(
  input,
  output_size
)[source]

Apply the 1d adaptive average pooling to input.

Parameters:
  • input (dragon.vm.torch.Tensor) – The input tensor.
  • output_size (Union[int, Sequence[int]]) – The target output size.
Returns:

dragon.vm.torch.Tensor – The output tensor.