Method shape

LanguageNetwork/BERT/models/neural.py:142–145

projection

(x)

        query_len = query.size(1)

        def shape(x):
            """  projection """
            return x.view(batch_size, -1, head_count, dim_per_head) \
                .transpose(1, 2)

        def unshape(x):
            """  compute context """