table-to-csv-pipeline / models /modules /sample_torch_module.py
rizgiak's picture
initial commit
4df4988
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