- 博客(2)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 2021-07-28
Torch.tensor.eq() 返回张量维度描述：分类问题中，常用eq()函数来统计正确样本个数，若张量维度不严格相同时，返回的矩阵维度会发生变化例：a与b同为1维张量（shape=3），c为2维张量（shape=[3,1]）a = torch.tensor([1,2,3])b = torch.tensor([1,2,2])c = torch.tensor([[1],[2],[3]])1. 维度相同a与b维度相同，eq()函数返回维度相同的张量（shape=3），可用于后续计数>
2021-07-28 21:12:20
77
原创 2021-07-27
Pytorch Geometric 经典模型https://pytorch-geometric.readthedocs.io/en/latest/modules/nn.html#modelsGCNThe Graph Neural Network from the “Semi-supervised Classification with Graph Convolutional Networks” paper, using the GCNConv operator for message passing.
2021-07-27 19:22:12
200
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人