AvgPool2d

Average pool 2d

Downsamples the input, returning the average for each pool/window.

Constructors

AvgPool2d

fun AvgPool2d(poolHeight: Int, poolWidth: Int)

cpu

open override fun cpu(): Layer<AvgPool2d>

equals

open operator override fun equals(other: Any?): Boolean

getSingleInput

Helper to check that the layer was called with a single input. Returns that input if successful, else errors.

gpu

open override fun gpu(): Layer<AvgPool2d>

hashCode

open override fun hashCode(): Int

invoke

open operator override fun invoke(input: DTensor): DTensor

open operator override fun invoke(vararg inputs: DTensor): DTensor

open fun to(device: Device): OnDevice

poolHeight

poolWidth