Skip to content

Inconsistent API between Maxpooling and Averagepooling #1819

Open
@zhichao-li

Description

@zhichao-li
object SpatialMaxPooling extends ModuleSerializable {
  def apply[@specialized(Float, Double) T: ClassTag](
      kW: Int,
      kH: Int,
      dW: Int = 1,
      dH: Int = 1,
      padW: Int = 0,
      padH: Int = 0,
      format: DataFormat = DataFormat.NCHW)
object SpatialAveragePooling {
  def apply[T: ClassTag](
      kW: Int,
      kH: Int,
      dW: Int = 1,
      dH: Int = 1,
      padW: Int = 0,
      padH: Int = 0,
      globalPooling: Boolean = false,
      ceilMode: Boolean = false,
      countIncludePad: Boolean = true,
      divide: Boolean = true,
      format: DataFormat = DataFormat.NCHW)

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions