畳み込みを式で表すと以下のようになる。Lはフィルタ数、Kは入力のチャネル数、xは入力データ、wは重みを表す。
$$h_{i,j}^{l} = \Sigma_{p,q,k}x_{i+p,j+q}^{k}w_{p,q}^{l}$$
図式化と実装アイデアを書いたものが以下です。
forループを減らしたい....
Register as a new user and use Qiita more conveniently