| import torch | |
| import torch.nn as nn | |
| class SensorEncoder(nn.Module): | |
| def __init__(self): | |
| super().__init__() | |
| self.fc = nn.Linear(10, 32) | |
| def forward(self, x): | |
| return self.fc(x) | |
| def main(): | |
| model = SensorEncoder() | |
| dummy_input = torch.randn(1, 10) | |
| output = model(dummy_input) | |
| print(output) | |
| if __name__ == "__main__": | |
| main() | |