(self, x)
| 615 | nn.init.zeros_(self.proj.weight) |
| 616 | |
| 617 | def forward(self, x): |
| 618 | x = self.proj(x) |
| 619 | return rearrange(x, "... h w (nh nw e) -> ... (h nh) (w nw) e", nh=self.h, nw=self.w) |
| 620 | |
| 621 | class TokenSplit(nn.Module): |
| 622 | def __init__(self, in_features, out_features, patch_size=(2, 2)): |
nothing calls this directly
no outgoing calls
no test coverage detected