Dropout.mdx

import { Callout } from "nextra-theme-docs" import { Tab, Tabs } from "nextra-theme-docs"

Dropout

A dropout layer that wraps a given Component.

Dropout(nn.Linear(128, 256), p=0.2)

Forward

(x: jaxtyping.Float[Tensor, ' *shape']) -> jaxtyping.Float[Tensor, ' *shape']