将输入数据input与权重矩阵self.weight做矩阵乘法,得到支持矩阵(support matrix)用于后续的计算。其中input的尺寸为(batch_size, input_dim),self.weight的尺寸为(input_dim, output_dim),得到的support的尺寸为(batch_size, output_dim)。

def forwardself input adj support = torchmminput selfweight output = torchspmmadj support if selfbias is not None return output + selfbias else return

原文地址: https://www.cveoy.top/t/topic/eHfc 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录