FeatureDropoutΒΆ
In-place version of FeatureDropout.
Abstract Signature:
FeatureDropout_(input: Tensor, p: float = 0.5, training: bool = False)
JAX (Core)
API:
βStrategy: Custom / Partial
In-place version of FeatureDropout.
Abstract Signature:
FeatureDropout_(input: Tensor, p: float = 0.5, training: bool = False)
β