(self, init_features: Tensor)
| 118 | self.add_module('denselayer%d' % (i + 1), layer) |
| 119 | |
| 120 | def forward(self, init_features: Tensor) -> Tensor: |
| 121 | features = [init_features] |
| 122 | for name, layer in self.items(): |
| 123 | new_features = layer(features) |
| 124 | features.append(new_features) |
| 125 | return torch.cat(features, 1) |
| 126 | |
| 127 | |
| 128 | class _Transition(nn.Sequential): |
nothing calls this directly
no outgoing calls
no test coverage detected