dropout_op

→TensorImplPtr

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

Element-wise inverted dropout — free-function entry point.

Thin wrapper that delegates to DropoutBackward::forward.

aTensorImplPtr

Input tensor of any shape.

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…