(self, x)
| 182 | self.r = r |
| 183 | |
| 184 | def forward(self, x): |
| 185 | out = self.conv(x) |
| 186 | batch_size, nchannels, H, W = out.shape |
| 187 | out = out.view((batch_size, self.r, nchannels // self.r, H, W)) |
| 188 | out = out.permute(0, 2, 3, 4, 1) |
| 189 | out = out.contiguous().view((batch_size, nchannels // self.r, H, -1)) |
| 190 | return out |
| 191 | |
| 192 | |
| 193 | class DenseBlock(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected