您即将离开知乎，请注意您的账号和财产安全。
https://discuss.pytorch.org/t/selecting-element-on-dimension-from-list-of-indexes/36319/3
