pytorch
文章平均质量分 61
MG冰淇淋
这个作者很懒,什么都没留下…
展开
-
Pytorch中gather函数的个人理解方法
之前一直理解不了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原创 2021-06-28 16:32:04 · 426 阅读 · 0 评论 -
conda安装pytorch遇到的坑
**一、根据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原创 2020-09-21 18:26:49 · 16139 阅读 · 3 评论