Pytorch-geomretic学习笔记 params_count,sparseTensor
params_count(model),model是nn.Module类型的,计算参数的数量。SparseTensor:rom torch_sparse import SparseTensoradj = SparseTensor(row=edge_index[0], col=edge_index[1], value=..., sparse_sizes=(num_nodes, num_nodes))# value is optional and can b
原创
2022-04-12 10:10:19 ·
1614 阅读 ·
0 评论