import torch import torch.nn as nn import torch.nn.functional as F class SensorExpert(nn.Module): def __init__(self): super(SensorExpert, self).__init__() self.fc1 = nn.Linear(10, 128) def forward(self, x): x = F.relu(self.fc1(x)) return x