import torch
from torch import nn
from torch.profiler import ProfilerActivity, profile, record_function


torch.manual_seed(7)

model = nn.Sequential(
    nn.Linear(8, 16),
    nn.ReLU(),
    nn.Linear(16, 4),
)
model.eval()
inputs = torch.randn(32, 8)