61992ce
1
2
3
4
5
6
7
8
9
10
import torch import torch.nn as nn from .. import relu class ReLU(nn.Module): def forward(self, x: torch.Tensor) -> torch.Tensor: return relu(x)
