src/transformers/models/swin/modeling_tf_swin.py/TFSwinMainLayer/get_head_mask
class TFSwinMainLayer:    def get_head_mask(self, head_mask: Optional[Any]) -> List:
        if head_mask is not None:
            raise NotImplementedError
        return [None] * len(self.config.depths)