tensor/张量
什么是tensor?
张量是用来探究一个点在各个切面(一共三个切面)和各个方向(x,y,z三个方向)上的受力情况。
所以一共有九种情况,可以用一个3x3的矩阵进行存储。
探究长方体内某一个节点的受力情况
如何用代码实现tensor
tensor的应用
tensor在图神经网络中的应用
用来构造图的Data要求使用tensor的结构进行传入,需要用tensor去构造点的属性,点的标签和边集等等。
import torch
from torch_geometric.data import Data
x = torch.tensor(([1,0],[0,1],[-1,0],[0,-1]),dtype=torch.float)
y = torch.tensor([0,1,0,1],dtype=torch.float)
edge_index = torch.tensor([[0,0,1,3,2],
[2,1,0,2,1]],dtype=torch.long)#前一个列表,代表起始点
#后一个列表代表终止点
data = Data(x=x,y=y,edge_index=edge_index)
print(data)