272123ed029d2e8f73e83d1d9ab1a96c56cead15,tensorflow_encrypted/tensor/prime.py,PrimeTensor,add,#PrimeTensor#Any#,69

Before Change

        return self.mod(k)

    def add(self, other: Union["PrimeTensor", int]) -> "PrimeTensor":
        x, y = _lift(self, self.modulus), _lift(other, self.modulus)
        return PrimeTensor((x.value + y.value) % self.modulus, self.modulus)

    def sub(self, other: Union["PrimeTensor", int]) -> "PrimeTensor":
        x, y = _lift(self, self.modulus), _lift(other, self.modulus)

After Change

        return self.mod(k)

    def add(self, other) -> "PrimeTensor":
        x, y = _lift(self, other)
        return self.factory.tensor((x.value + y.value) % self.modulus)

    def sub(self, other) -> "PrimeTensor":
        x, y = _lift(self, other)

Instances