Where is the documentation for torch.nn.modules.conv.Conv2d?

Making an instance of torch.nn.Conv2d returns an object ot type torch.nn.modules.conv.Conv2d.

Somehow I can’t find the documentation for that class anywhere (I found this ).

Thanks in advance.

torch.nn.Conv2d is just a reference to torch.nn.modules.conv.Conv2d. To understand that here are the details from the source code.

torch.nn module imports * from .modules
torch.nn.modules imports Conv2d from .conv

Therefore the docs for torch.nn.Conv2d are the ones you are looking for.

1 Like