File size: 112 Bytes
f701ae7
 
 
 
 
 
 
1
2
3
4
5
6
7
8
import torch

from ._ops import ops


def relu(input: torch.Tensor) -> torch.Tensor:
    return ops.relu(input)