I checked the data flow after you replied. I found it became NaN between the second and the third layer, i.e.between the relu function in self.l2 and unit_gcn layer in self.l3. However, it confused me because there is no operation between them. The mode only assigned the output of self.relu in self.l2 to x, and then x became NaN. It’s really confusing.