d51512dcf397a6b6a70fe46885f1b5acd45205b3,scattering/scattering2d/utils.py,Modulus,__call__,#Modulus#Any#,110

Before Change

    def __call__(self, input):
        if not self.jit or not input.is_cuda:
            norm = input.norm(p=2, dim=-1, keepdim=True)
            return torch.cat([norm, norm.new(norm.size()).zero_()], -1)

        out = input.new(input.size())
        input = input.contiguous()

After Change

    def __call__(self, input):

        if not input.is_cuda and self.backend=="skcuda":
            raise RuntimeError("Use the torch backend for cpu tensors!")

        if self.backend=="torch":
            norm = input.norm(p=2, dim=-1, keepdim=True)
            return torch.cat([norm, torch.zeros_like(norm)], -1)

Instances