Method __init__

models/aios/position_encoding.py:13–26

(self,
                 num_pos_feats=64,
                 temperature=10000,
                 normalize=False,
                 scale=None)

    to the one used by the Attention is all you need paper, generalized to work
    on images."""
    def __init__(self,
                 num_pos_feats=64,
                 temperature=10000,
                 normalize=False,
                 scale=None):
        super().__init__()
        self.num_pos_feats = num_pos_feats
        self.temperature = temperature
        self.normalize = normalize
        if scale is not None and normalize is False:
            raise ValueError('normalize should be True if scale is passed')
        if scale is None:
            scale = 2 * math.pi
        self.scale = scale

    def forward(self, tensor_list: NestedTensor):
        x = tensor_list.tensors