def forward(self, x):
        """
        Args:
            x (torch.Tensor): batch of data, expected shapes [B, 3, H, W]

        Returns:
            bbox locations (torch.Tensor) with shapes [B, A, 4],
                where B - batch size, A - num anchors
            class confidence logits (torch.Tensor) with shapes [B, A, N_CLASSES],
                where B - batch size, A - num anchors
        """
        x = self.feature_extractor(x)

        detection_feed = [x]
        for layer in self.additional_blocks:
            x = layer(x)
            detection_feed.append(x)

        # Feature Map 38x38x4, 19x19x6, 10x10x6, 5x5x6, 3x3x4, 1x1x4
        locs, confs = self.bbox_view(detection_feed, self.loc, self.conf)

        return locs, confs

        Args:
            x (torch.Tensor): batch of data, expected shapes [B, 3, H, W]

        Returns:
            bbox locations (torch.Tensor) with shapes [B, A, 4],
                where B - batch size, A - num anchors
            class confidence logits (torch.Tensor) with shapes [B, A, N_CLASSES],
                where B - batch size, A - num anchors
        
examples/detection/models/ssd.py