fn
dropoutnd_op
→TensorImplPtrint dropoutnd_op(const int & a, double p, bool training, Generator * gen)Channel-wise dropout — free-function entry point.
Thin wrapper that delegates to DropoutNdBackward::forward.
Handles 3-D, 4-D, and 5-D inputs uniformly (assumes the layout
).
Parameters
aTensorImplPtrInput tensor of rank in layout .
pdoubleChannel drop probability in .
trainingboolIf
false, the forward is a pass-through.genGenerator*Optional RNG;
nullptr selects default_generator().Returns
TensorImplPtrOutput tensor of the same shape as a.