vkML 0.1.0

ReLU

Apply the rectified linear unit, element-wise.

class ReLU : Modulepython/vkml/nn.py:304

A stateless wrapper: forward calls relu and nothing else. It exists so the activation can sit inside a Sequential. See relu for the numerical behaviour, which is where the substance is.

Forward

forward

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

See also relu, Sequential

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