https://github.com/waterdisappear/ATRNet-STAR/blob/6c6492bd841b3ec954b01174f2a3d978adc60178/ATRBench/Classification/SARatrX/model/HiVit.py#L342C10-L342C30