Einsum

パブリック最終クラスEinsum

2 入力 1 出力の基本的な einsum 演算をサポートする演算です。

この操作は、tf.einsum のように明示的に再形成および転置操作を行わないため、TPU パフォーマンスが向上します。

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends TType > Einsum <T>
create (スコープscope、オペランド<T> a、オペランド<T> b、文字列式)
新しい Einsum オペレーションをラップするクラスを作成するファクトリ メソッド。
出力<T>
製品()

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "XlaEinsum"

パブリックメソッド

public Output <T> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static Einsum <T> create (スコープscope、オペランド<T> a、オペランド<T> b、文字列式)

新しい Einsum オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
戻り値
  • Einsum の新しいインスタンス

public出力<T> product ()