I am trying to execute my dataset with batch size = 100. It works fine with batch size = 1 but with 100, it gives the error as: ValueError: Expected input batch_size (1) to match target batch_size (100).
 class Net(nn.Module):
        def __init__(self):
            super(Net, self).__init__()
            self.conv1 = nn.Conv1d(100, 1, kernel_size=1)
            self.dropout = nn.Dropout2d()
            self.fc1 = nn.Linear(32, 16)
            self.fc2 = nn.Linear(16, 1)
            self.hybrid = Hybrid(qiskit.Aer.get_backend('qasm_simulator'), 100, np.pi / 2)
    
        def forward(self, x):
            x = F.relu(self.conv1(x))
            x = self.dropout(x)
            x = x.view(x.size(0), -1)
            x = F.relu(self.fc1(x))
            x = self.fc2(x)
            x = self.hybrid(x)
            return torch.cat((x, 1 - x), -1)
    ```
    The sizes of data, label and output where output = model(data) are
    torch.Size([100, 32])
    torch.Size([100])
    torch.Size([1, 2])
    ```
    How to make the size of output same as data size?