alpha_dropout_op

→TensorImplPtr

int alpha_dropout_op(const int & a, double p, bool training, Generator * gen)

SELU-preserving alpha dropout — free-function entry point.

Thin wrapper that delegates to AlphaDropoutBackward::forward.

aTensorImplPtr

Input tensor (typically the output of a SELU activation).

pdouble

Drop probability in

[0, 1)

trainingbool

If false, the forward is a pass-through.

genGenerator*

Optional RNG; nullptr selects default_generator().

TensorImplPtr

Output tensor of the same shape as a.

Loading…