Like for example from “2d” layers like these:
Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)

Does PyTorch have any way to extract the information that I desire?