예제 #1
0
 def positive(self):
     # self >= 0
     return securenn.relu_deriv(self)
예제 #2
0
def test_relu_deriv(workers):
    alice, bob, james = workers["alice"], workers["bob"], workers["james"]
    x = th.tensor([10, 0, -3]).share(alice, bob, crypto_provider=james).child
    r = relu_deriv(x)

    assert (r.get() == th.tensor([1, 1, 0])).all()