2021SC@SDUSC
本文将针对verorize中relfix函数、getEnts函数、listToBatch函数进行分析
重点在于三者之间的关联性分析。
首先对relfix函数进行分析
mat = []
for x in relstrs:
pieces = x.strip().split(';')
x = [[int(y)+len(self.REL.special) for y in z.strip().split()] for z in pieces]
mat.append(torch.LongTensor(x).cuda())
同样,初始化局部变量值,创建了一个空列表mat,用于存放追加的值
遍历传入的relstrs,即reverse后的字符串列表,按照;分割,得到pieces局部变量值
接下来,采用了二次遍历直接赋值的方式,从内层开始,获取到每个piece在z方向上的向量,再通过z分解得到y向量,将y向量于特殊字段长相加得到x。
根据x值调用cuda生成多维矩阵张量,并追加至mat列表。
lens = [x.size(0) for x in mat]
m = max(lens)
mat = [self.pad(x,m) for x in mat]
mat