in model/interpolation_net.py [0:0]
def rot_sub(self, shape, r):
if shape.sub is not None:
for i_p in range(len(shape.sub[0])):
shape.sub[0][i_p][0, :, :] = torch.mm(shape.sub[0][i_p][0, :, :], r)
if shape.vert_full is not None:
shape.vert_full = torch.mm(shape.vert_full, r)
return shape