pytorchのconv2dについて

Question

pytorchで実装できる畳み込み層でconv2dというのがありますが、これの出力がわかりません。

例えば、

self.conv1 = nn.Conv2d(1, 6, 3)

であれば、どのような出力がえられるのかを教えて欲しいです。

ちなみに、pytorch関連のサイトを読んだのですがいまいち理解ができていないです。

Answer 1

self.conv1 = nn.Conv2d(1, 6, 3)

これは畳み込み層のオブジェクトを作っているだけです。オブジェクトを作っているだけなので、データを渡して出力を得るというコードではありません。
引数の意味は入力チャンネル数が1、出力チャンネル数が6、カーネルサイズが3です。

Conv2d — PyTorch 1.7.0 documentation

torch.nn.Conv2d(in_channels: int, out_channels: int, kernel_size: Union[T, Tuple[T, T]])