max_pool2d

dragon.vm.tensorflow.nn.max_pool2d(
  input,
  ksize,
  strides,
  padding,
  data_format='NHWC',
  name=None
)[source]

Apply the 2d max pooling.

Parameters:
  • input (dragon.Tensor) – The input tensor.
  • ksize (Sequence[int]) – The size(s) of sliding window.
  • strides (Sequence[int], optional) – The stride(s) of sliding window.
  • padding (Union[{'valid', 'same'}, Sequence[int]]) – The padding algorithm or padding size(s).
  • data_format ({'NCHW', 'NCDHW', 'NHWC', 'NDHWC'}, optional) – The optional data format.
  • name (str, optional) – A optional name for the operation.
Returns:

dragon.Tensor – The output tensor.