Write pdb.set_trace() inside a for loop.

screenshot

import torch from torch import nn import pdb

for i in range(1): pdb.set_trace() linear = nn.Linear(3, 2)

Enter s and find the output as below: ipdb> s --Call--

/usr/local/lib/python3.11/dist-packages/torch/nn/modules/linear.py(93)init()