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 June 7, 2023, 6:19 p.m.