ag_dropout: Create a Dropout layer

View source: R/ag_layers.R

ag_dropoutR Documentation

Create a Dropout layer

Description

In training mode applies inverted dropout (random Bernoulli mask, scale by 1/(1-rate) to preserve expected values). In eval mode is identity.

Usage

ag_dropout(rate)

Arguments

rate

Drop probability in [0, 1)

Value

An ag_dropout environment

Examples


drop <- ag_dropout(0.5)
x    <- ag_tensor(matrix(runif(8), 4, 2))
out  <- drop$forward(x)  # training mode by default
ag_eval(drop)
out2 <- drop$forward(x)  # identity


ggmlR documentation built on July 14, 2026, 1:08 a.m.