table-extraction-from-image / models /modules /sample_torch_module.py
jurgendn's picture
[Demo Gradio] Add to Huggingface Space
4c41a36
raw
history blame
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