def func(self, input_tensor: Tensor) -> Tensor: copy_tensor = input_tensor.copy() copy_tensor[copy_tensor <= 0] = 0 return copy_tensor
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
def grad(X: Tensor) -> Tensor: copy_X = X.copy() copy_X[copy_X < 0] = 0 copy_X[copy_X != 0] = 1 return copy_X