NativeDropoutΒΆ
Native dropout implementation.
Abstract Signature:
NativeDropout(input: Tensor, p: float, train: bool)
JAX (Core)
API:
jax.random.bernoulliStrategy: Plugin (dropout_macro)
Native dropout implementation.
Abstract Signature:
NativeDropout(input: Tensor, p: float, train: bool)
jax.random.bernoulli