TensorFlowのカスタム空間畳込み

Question

TensorFlowのtf.nn.conv2d関数を、フィルターを乗算する前に各受容野内の入力（シグモイドまたはReLU関数）をしきい値にして拡張したいと考えています。実際の畳み込み演算はEigenで実装され、Eigenの畳み込み関数はchooseという関数にあります。しかし、私はchoose関数の実装を見つけることができませんでした。

TensorFlow githubリポジトリにあるchoose関数はどこにありますか？入力処理を行うためにEigenの畳み込み関数を拡張する簡単な方法はありますか？

ありがとうございます！

Answer 1

Eigenは、行優先および列主メモリレイアウトの両方をサポートしています。 「選択」テンプレートは、入力データのレイアウトに応じて適切な畳み込みコードパスを選択します。テンプレートそのものはhttp://eigen.tuxfamily.org/dox-devel/unsupported/TensorMeta_8h_source.html

で実際に畳み込みが行われます。実際の畳み込みは、畳み込み重み（別名カーネル）と入力画像から抽出された2dパッチの間の行列乗算（縮小と呼ばれる）として行われます。