This document is relevant for: Trn1, Trn2, Trn3

nki.language.relu#

nki.language.relu(x, dtype=None)[source]#

ReLU activation, element-wise.

This document is relevant for: Trn1, Trn2, Trn3