pytorch

61f05787 - Update take_along_dim docs to include `dim=None` case (#109120)

Update take_along_dim docs to include `dim=None` case (#109120) Pull Request resolved: https://github.com/pytorch/pytorch/pull/109120 Approved by: https://github.com/lezcano ghstack dependencies: #108879, #108880