テンソルフロー::作戦:: 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