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.
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.