インテル® DAAL 2018 デベロッパー・ガイド
フォワード 2 次元局所結合層は、入力引数 x にサイズ m1 x m2 の nKernels 2 次元カーネル K のセットを適用して、値テンソル Y を計算します。ライブラリーは、4 次元の入力テンソル X ∈ Rn1 x n2 x n3 x n4 をサポートします。そのため、次の式が適用されます。
ここで、i + a < n1, j + b < n2、および r はカーネル・インデックスです。
カーネルのセットは、入力引数 x の選択された次元に固有です。
2 次元局所結合層の詳細は、[GregorLecun2010] を参照してください。
一般性を失うことなく、畳み込みカーネルが最後の 2 つの次元に適用されると仮定します。
以下の項目を指定します。
入力データを含む 4 次元のテンソル X ∈ Rn1 x n2 x n3 x n4
カーネルのパラメーター/重みを含む 6 次元のテンソル K ∈ RnKernels x l3 x l4 x m2 x m3 x m4
各カーネルのバイアスを含む 3 次元のテンソル B ∈ RnKernels x l3 x l4
上記のテンソルについて:
です。pi はそれぞれのパディングです。
nGroups は次のように定義されます: n2 がグループ次元であると仮定します。入力テンソルはこの次元で nGroups グループに分割され、値および重みのテンソルは nKernels 次元で nGroups グループに分割されます。nKernels および n2 は nGroups の倍数でなければなりません。値の各グループは、入力データ、重み、バイアスのテンソルのグループを使用して計算されます。
次のような値の 4 次元のテンソル Y ∈ Rn1 x nKernels x l3 x l4 を計算します。
ここで、
s3 および s4 はストライド