Example #1
0
 def func(self, input_tensor: Tensor) -> Tensor:
     copy_tensor = input_tensor.copy()
     copy_tensor[copy_tensor <= 0] = 0
     return copy_tensor
Example #2
0
 def grad(self, input_tensor: Tensor) -> Tensor:
     copy_tensor = input_tensor.copy()
     copy_tensor[copy_tensor <= 0] = 0
     copy_tensor[copy_tensor > 0] = 1
     return copy_tensor
Example #3
0
 def grad(X: Tensor) -> Tensor:
     copy_X = X.copy()
     copy_X[copy_X < 0] = 0
     copy_X[copy_X != 0] = 1
     return copy_X