nnf_prelu: Prelu

View source: R/nnf-activation.R

nnf_preluR Documentation

Prelu

Description

Applies element-wise the function PReLU(x) = max(0,x) + weight * min(0,x) where weight is a learnable parameter.

Usage

nnf_prelu(input, weight)

Arguments

input

(N,*) tensor, where * means, any number of additional dimensions

weight

(Tensor) the learnable weights


torch documentation built on May 29, 2024, 9:54 a.m.