随随便便记录些东西

之前一直理解不了Pytorch中gather的用法，看了官方的文档也是一头雾水。然后自己琢磨，找规律，用以下方法进行理解。 一、官方文档 torch.gather(input, dim, index, out=None) → Tensor Gathers values along an axis specified by dim. For a 3-D tensor the output is specified by: out[i][j][k] = input[index

** 一、根据cuda选择自己合适的pytorch版本 ** https://pytorch.org/get-started/locally/ conda install pytorch torchvision cudatoolkit=10.2 -c pytorch 二、配置清华镜像 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channel