avgPool: Average pooling layer

In cito: Building and Training Neural Networks

Average pooling layer

Description

creates a 'avgPool' 'citolayer' object that is used by create_architecture.

Usage

avgPool(kernel_size = NULL, stride = NULL, padding = NULL)

Arguments

kernel_size	(int or tuple) size of the kernel in this layer. Use a tuple if the kernel size isn't equal in all dimensions
stride	(int or tuple) stride of the kernel in this layer. NULL sets the stride equal to the kernel size. Use a tuple if the stride isn't equal in all dimensions
padding	(int or tuple) zero-padding added to both sides of the input. Use a tuple if the padding isn't equal in all dimensions

Details

This function creates a 'avgPool' 'citolayer' object that is passed to the create_architecture function. The parameters that aren't assigned here (and are therefore still NULL) are filled with the default values passed to create_architecture.

Value

S3 object of class "avgPool" "citolayer"

Author(s)

Armin Schenk

See Also

create_architecture

cito documentation built on May 29, 2024, 1:28 a.m.