バッチ化された ND `SparseTensor` にソフトマックスを適用します。
入力は、論理形状 `[..., B, C]` (ここで `N >= 2`) を持ち、標準辞書順でソートされたインデックスを持つ ND SparseTensor を表します。
この操作は、通常のtf.nn.softmax()
形状 `[B, C]` を持つ各最も内側の論理部分行列に適用するのと同等ですが、暗黙的にゼロの要素が関与しないという点が異なります。具体的には、アルゴリズムは次と同等です。
(1) tf.nn.softmax()
サイズ C 次元に沿った、形状 `[B, C]` を持つ最も内側の各部分行列の高密度ビューに適用します。 (2) 元の暗黙的にゼロの位置をマスクアウトします。 (3) 残りの要素を繰り込みます。
したがって、「SparseTensor」の結果は、まったく同じ非ゼロのインデックスと形状を持ちます。
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T extends TNumber > SparseSoftmax <T> | |
出力<T> | 出力() 1-D。 |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static SparseSoftmax <T> create (スコープscope、オペランド<TInt64> spIndices、オペランド<T>spValues、オペランド<TInt64> spShape)
新しい SparseSoftmax オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
spインデックス | 2D。 SparseTensor 内の空でない値のインデックスを正規の順序で持つ `NNZ x R` 行列。 |
sp値 | 1-D。 `NNZ` は `sp_indices` に対応する空ではない値です。 |
spシェイプ | 1-D。入力 SparseTensor の形状。 |
戻り値
- SparseSoftmax の新しいインスタンス