https://reviewsvideo.ru/en/videos/lWsswhcoWO0/torch-func-functional-transforms-in-pytorch