nnf_alpha_dropout: Alpha_dropout

View source: R/nnf-dropout.R

nnf_alpha_dropoutR Documentation

Alpha_dropout

Description

Applies alpha dropout to the input.

Usage

nnf_alpha_dropout(input, p = 0.5, training = FALSE, inplace = FALSE)

Arguments

input

the input tensor

p

probability of an element to be zeroed. Default: 0.5

training

apply dropout if is TRUE. Default: TRUE

inplace

If set to TRUE, will do this operation in-place. Default: FALSE


torch documentation built on June 7, 2023, 6:19 p.m.