(self, x, t)
| 94 | ) |
| 95 | |
| 96 | def forward(self, x, t): |
| 97 | x = self.maxpool_conv(x) |
| 98 | emb = self.emb_layer(t)[:, :, None, None].repeat(1, 1, x.shape[-2], x.shape[-1]) |
| 99 | return x + emb |
| 100 | |
| 101 | |
| 102 | class Up(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected