skainet-lang-core/sk.ainet.lang.nn.dsl/AvgPool2dImpl

AvgPool2dImpl

class AvgPool2dImpl<T : DType, V>(val executionContext: ExecutionContext, initialKernelSize: Pair<Int, Int>, initialStride: Pair<Int, Int>, initialPadding: Pair<Int, Int>, initialCountIncludePad: Boolean, id: String) : AVGPOOL2D<T, V> (source)

Constructors

constructor(executionContext: ExecutionContext, initialKernelSize: Pair<Int, Int>, initialStride: Pair<Int, Int>, initialPadding: Pair<Int, Int>, initialCountIncludePad: Boolean, id: String)

Properties

countIncludePad

open override var countIncludePad: Boolean

executionContext

open override val executionContext: ExecutionContext

open override var kernelSize: Pair<Int, Int>

open override var padding: Pair<Int, Int>

open override var stride: Pair<Int, Int>

Functions

fun create(): AvgPool2d<T, V>

open override fun kernelSize(size: Int)

open override fun padding(size: Int)

open override fun stride(size: Int)