たまにconv1d,2dを使おうとすると毎回次元数の関係性を忘れてしまうので備忘録で。
次の図のようにすれば出力のチャネル数は任意に設定できる
conv1dの場合は次のような感じ
ついでにpytorchで確認
conv1dの引数はそれぞれ、入力チャネル数、出力チャネル数、カーネルサイズ
import torch from torch import nn conv1d = nn.Conv1d(5, 3, 8) print(conv1d.weight.shape) # torch.Size([3, 5, 8]) # 入力(最初はバッチサイズ) x = torch.rand(4, 5, 10) y = conv1d(x) print(y.shape) # torch.Size([4, 3, 3])