Pytorch
文章平均质量分 84
记录深度学习pytorch中的相关知识点儿
steelDK
平平无奇,躺平青年;基础知识总是学了就忘,借用平台记录一下自己科研中可能涉及到的一些知识点儿,也希望能帮到有缘人!
展开
-
PyTorch Geometric (PyG) 库的安装
最近在学习图神经网络,需要用到PyG库,发现这个库的安装不能简单的使用pip install 安装,这里记录一下。原创 2023-05-11 11:28:53 · 5401 阅读 · 4 评论 -
Pytorch教程之torch.mm、torch.bmm、torch.matmul、masked_fill
这几天正在看NLP中的注意力机制,代码中涉及到了一些关于张量矩阵乘法和填充一些代码,这里积累一下。主要参考了pytorch2.0的官方文档。原创 2023-03-28 16:27:12 · 1982 阅读 · 0 评论 -
Pytorch教程之张量
Tensor中文翻译张量,是一个词不达意的名字。张量在不同学科中有不同的意义,在深度学习中张量表示的是一个多维数组,它是标量、向量、矩阵的拓展。标量是零维张量,向量是一维张量,矩阵是二维张量。tensor之于pytorch等同于ndarray之于numpy,它是pytorch中最核心的数据结构,用于表达各类数据,如输入数据、模型的参数、模型的特征图、模型的输出等。各类张量的api如下:可以参考。原创 2023-03-16 15:45:35 · 1292 阅读 · 0 评论