train.py

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()