I want to check the output value of each layer I build. but I didn’t find any element in layers. Is there any way to do this?
For example, I have a model:
model = nn.Sequential(nn.Linear(128, 64), nn.Linear(64, 32), nn.Linear(32, 1))
av = Variable(torch.rand(7, 128), requires_grad = True)
model.forward(av)

I want to see the output of each linear layer.