File size: 259 Bytes
4c41a36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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
|