이미지같은 큰 차원에 유용 github.comhttps://github.com/google/trax/blob/4d99ad4965bab1deba227539758d59f0df0fef48/trax/layers/research/position_encodings.py#L29
