fab47fcca9887402f7474c80545311fcb952566f,torch_geometric/transform/random_shear.py,RandomShear,__call__,#RandomShear#Any#,8

Before Change

        range = torch.arange(0, dim, out=torch.LongTensor())
        shear[range, range] = 1

        shear = shear.type_as(pos)
        pos = torch.matmul(shear, pos.view(-1, dim, 1))
        pos = pos.view(-1, dim)

        pos += mean

After Change

        matrix = data.pos.new_empty(dim, dim).uniform(-self.scale, self.scale)
        eye = torch.arange(dim, dtype=torch.long)
        matrix[eye, eye] = 1

        return LinearTransformation(matrix)(data)

Instances