构建自己的图数据集

代码:

python 复制代码
import warnings
warnings.filterwarnings("ignore")
import torch
from torch_geometric.data import Data

x = torch.tensor([[2,1],[5,6],[3,7],[12,0]],dtype=torch.float)
y = torch.tensor([0,1,0,1],dtype=torch.float)

#定义边
edge_index = torch.tensor([[0,1,2,0,3], #起始点
                           [1,0,1,3,2]],dtype=torch.long) #终止点


data = Data(x=x,edge_index=edge_index,y=y)
print(data)

TopKPooling流程:

对图片进行剪枝操作,选择分低的节点剔除掉,然后再重新组合成一个新的图

图注意力机制与序列图模型:

序列图神经网络(GNN)

相关推荐
晚烛3 分钟前
CANN + 物理信息神经网络(PINNs):求解偏微分方程的新范式
javascript·人工智能·flutter·html·零售
爱吃烤鸡翅的酸菜鱼5 分钟前
CANN ops-math向量运算与特殊函数实现解析
人工智能·aigc
波动几何16 分钟前
OpenClaw 构建指南:打造智能多工具编排运行时框架
人工智能
程序猿追17 分钟前
深度解码AI之魂:CANN Compiler 核心架构与技术演进
人工智能·架构
新缸中之脑18 分钟前
Figma Make 提示工程
人工智能·figma
赫尔·普莱蒂科萨·帕塔19 分钟前
智能体工程
人工智能·机器人·软件工程·agi
觉醒大王21 分钟前
AI写的青基中了
人工智能·笔记·深度学习·学习·职场和发展·学习方法
深鱼~23 分钟前
深度剖析ops-transformer:LayerNorm与GEMM的融合优化
人工智能·深度学习·transformer
哈__25 分钟前
CANN图优化技术:深度学习模型的编译器魔法
人工智能·深度学习
灰灰勇闯IT27 分钟前
神经网络的基石——深度解析 CANN ops-nn 算子库如何赋能昇腾 AI
人工智能·深度学习·神经网络