- 博客(2)
- 收藏
- 关注
RSS订阅原创 torch_geometric.data.collate中collate（）函数
在进行图数据集预处理过程中，我们将每一个图数据存为一个Data类型的数据，一个数据集涉及大量的图，将大量的Data数据分开存储是非常慢的使用也不方便，因此需要用到collate（）
2022-12-15 10:11:57
919
1
原创 torch.cat()中dim说明
初次接触众多博客对dim的讲解为，对于两个二维张量作为输入，dim取0结果为两个张量按行拼接，取1结果为按列拼接，但是对于高维来说就有点难以直观想象结果了，我们尝试三维情况进而总结规律。其中初学者最费解的就是dim的选取，dim的取值范围由输入张量的维度决定，输入为n维张量，dim取值在[0,n-1],接下来我们以实验理解dim不同取值对应的不同操作结果。至此，torch.cat（）的dim作用已经讲清楚，建议动手实验一下就可以弄明白其中的奥秘！
2022-12-14 10:16:29
3196
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人