Skip to content

Latest commit

History

History
24 lines (18 loc) 路 351 Bytes

PreNorm.mdx

File metadata and controls

24 lines (18 loc) 路 351 Bytes

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

PreNorm

Apply Pre-Layer normalization to a children

PreNorm(
    nn.Sequential(
        nn.Linear(100, 200),
        nn.GELU(),
        nn.Linear(200, 100)
    ),
    dim=100
)

Forward

(x: torch.Tensor) -> torch.Tensor