File size: 259 Bytes
4df4988
 
 
 
 
 
 
 
 
 
 
 
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