facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
6.93k stars 1.17k forks source link

why this randmasking works in this implementation? #124

Closed FerryHuang closed 1 year ago

FerryHuang commented 1 year ago

I notice two "torch.argsort" function being used to the "noise" to produce the "ids_shuffe", but how can "ids_shuffle" unshuffle the x?