テンソルフロー::作戦:: FusedPadConv2D
#include <nn_ops.h>
畳み込み中に前処理としてパディングを実行します。
まとめ
FusedResizeAndPadConv2d と同様に、この演算により、空間パディング変換ステージが im2col ルックアップと融合される最適化された実装が可能になりますが、この場合、サイズ変更に必要な双線形フィルタリングは必要ありません。パディングを融合すると、中間結果をテンソル全体として書き出す必要がなくなり、メモリ負荷が軽減され、変換計算をマージすることでレイテンシをいくらか向上させることができます。 Conv2Dの data_format 属性はこの操作ではサポートされていないため、代わりに「NHWC」順序が使用されます。内部的には、この操作はグラフごとに 1 つのスクラッチ バッファーを使用します。これは、複数のバージョンが並行して実行されている場合にブロックされることを意味します。これは、この演算子が主にメモリ使用量を最小限に抑えるための最適化であるためです。
引数:
- スコープ:スコープオブジェクト
- 入力: 形状
[batch, in_height, in_width, in_channels]
の 4-D。 - paddings: パディング サイズを指定する 2 列の行列。行数は
input
のランクと同じである必要があります。 - フィルター: 形状
[filter_height, filter_width, in_channels, out_channels]
4-D。 - strides: 長さ 4 の 1-D。
input
の各次元のスライディング ウィンドウのストライド。 format で指定した次元と同じ順序である必要があります。 - padding: 使用するパディング アルゴリズムのタイプ。
戻り値:
-
Output
: 出力テンソル。
コンストラクターとデストラクター | |
---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
パブリック属性 | |
---|---|
operation | |
output |
公共機能 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
パブリック属性
手術
Operation operation
出力
::tensorflow::Output output
公共機能
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
ノード
::tensorflow::Node * node() const
演算子::tensorflow::入力
operator::tensorflow::Input() const
演算子::tensorflow::出力
operator::tensorflow::Output() const