FeatureDropoutΒΆ

In-place version of FeatureDropout.

Abstract Signature:

FeatureDropout_(input: Tensor, p: float = 0.5, training: bool = False)

PyTorch

API: torch.feature_dropout_
Strategy: Direct Mapping

JAX (Core)

API: β€”
Strategy: Custom / Partial