This document is relevant for: Trn2, Trn3

nki.language.prelu

nki.language.prelu = Ellipsis

Parametric ReLU activation function. Used as the op parameter in activation ISA instructions such as nki.isa.activation. The slope for negative inputs is controlled by the relu_param argument (see Supported Activation Functions for NKI ISA).

This document is relevant for: Trn2, Trn3