def forward(self, feats: Sequence[Tensor]) -> List[Tensor]:
        """Forward the network. The input is multi-stage feature maps and the
        output is a list of heatmaps from multiple stages.

        Args:
            feats (Sequence[Tensor]): Multi-stage feature maps.

        Returns:
            List[Tensor]: A list of output heatmaps from multiple stages.
        """
        out = []
        assert len(feats) == self.num_stages, (
            f'The length of feature maps did not match the '
            f'`num_stages` in {self.__class__.__name__}')
        for i in range(self.num_stages):
            y = self.multi_deconv_layers[i](feats[i])
            y = self.multi_final_layers[i](y)
            out.append(y)

        return out
Forward the network. The input is multi-stage feature maps and the
        output is a list of heatmaps from multiple stages.

        Args:
            feats (Sequence[Tensor]): Multi-stage feature maps.

        Returns:
            List[Tensor]: A list of output heatmaps from multiple stages.
        
mmpose/mmpose/models/heads/heatmap_heads/cpm_head.py