from torch import Tensor, nn | |
class UselessLayer(nn.Module): | |
def __init__(self) -> None: | |
super(UselessLayer, self).__init__() | |
self.seq = nn.Identity() | |
def forward(self, x: Tensor) -> Tensor: | |
x = self.seq(x) | |
return x | |
from torch import Tensor, nn | |
class UselessLayer(nn.Module): | |
def __init__(self) -> None: | |
super(UselessLayer, self).__init__() | |
self.seq = nn.Identity() | |
def forward(self, x: Tensor) -> Tensor: | |
x = self.seq(x) | |
return x | |