table-caption-extraction / models /modules /sample_torch_module.py
rizgiak's picture
initial commit
737c08b
raw
history blame contribute delete
259 Bytes
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