vkML 0.1.0

MaxPool2d

Maximum over each sliding window, per channel.

class MaxPool2d : Modulepython/vkml/nn.py:779

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

Construction

__init__

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

Forward

forward

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

Internals

__repr__

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

See also max_pool2d, AvgPool2d

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