vkML 0.1.0

AvgPool2d

Mean over each sliding window, per channel.

class AvgPool2d : Modulepython/vkml/nn.py:795

A stateless wrapper over avg_pool2d, holding only the window geometry.

Construction

__init__

def __init__self, kernel_size, stride=None, padding=0python/vkml/nn.py:796

Forward

forward

def forwardself, x: V.Tensor -> V.Tensorpython/vkml/nn.py:802

Internals

__repr__

def __repr__self -> strpython/vkml/nn.py:805

See also avg_pool2d, MaxPool2d

vkML — Vulkan-first machine learning in C++20. Apache-2.0. Signatures on this page are generated from the installed module.