Python tf.linalg.LinearOperatorComposition用法及代碼示例

組成一個或多個 LinearOperators 。

繼承自：LinearOperator，Module

用法

tf.linalg.LinearOperatorComposition(
    operators, is_non_singular=None, is_self_adjoint=None,
    is_positive_definite=None, is_square=None, name=None
)

該運算符組成一個或多個線性運算符 [op1,...,opJ] ，構建一個新的 LinearOperator ，其動作定義為：

op_composed(x):= op1(op2(...(opJ(x)...))

如果 opj 的行為類似於 [batch] 矩陣 Aj ，則 op_composed 的行為類似於由乘法 A1 A2...AJ 形成的 [batch] 矩陣。

如果 opj 具有形狀 batch_shape_j + [M_j, N_j] ，那麽我們必須具有 N_j = M_{j+1} ，在這種情況下，組合運算符的形狀等於 broadcast_batch_shape + [M_1, N_J] ，其中 broadcast_batch_shape 是 batch_shape_j , j = 1,...,J 的相互廣播，假設中間批次形狀播送。即使組合形狀被很好地定義，組合算子的方法也可能由於定義算子的方法中缺乏廣播能力而失敗。

# Create a 2 x 2 linear operator composed of two 2 x 2 operators.
operator_1 = LinearOperatorFullMatrix([[1., 2.], [3., 4.]])
operator_2 = LinearOperatorFullMatrix([[1., 0.], [0., 1.]])
operator = LinearOperatorComposition([operator_1, operator_2])

operator.to_dense()
==> [[1., 2.]
     [3., 4.]]

operator.shape
==> [2, 2]

operator.log_abs_determinant()
==> scalar Tensor

x = ... Shape [2, 4] Tensor
operator.matmul(x)
==> Shape [2, 4] Tensor

# Create a [2, 3] batch of 4 x 5 linear operators.
matrix_45 = tf.random.normal(shape=[2, 3, 4, 5])
operator_45 = LinearOperatorFullMatrix(matrix)

# Create a [2, 3] batch of 5 x 6 linear operators.
matrix_56 = tf.random.normal(shape=[2, 3, 5, 6])
operator_56 = LinearOperatorFullMatrix(matrix_56)

# Compose to create a [2, 3] batch of 4 x 6 operators.
operator_46 = LinearOperatorComposition([operator_45, operator_56])

# Create a shape [2, 3, 6, 2] vector.
x = tf.random.normal(shape=[2, 3, 6, 2])
operator.matmul(x)
==> Shape [2, 3, 4, 2] Tensor

性能

LinearOperatorComposition 在任何操作上的性能等於各個操作符操作的總和。

矩陣屬性提示

此 LinearOperator 使用 is_X 形式的布爾標誌初始化，用於 X = non_singular, self_adjoint, positive_definite, square 。它們具有以下含義：

• 如果 is_X == True ，調用者應該期望操作符具有屬性 X 。這是一個應該實現的承諾，但不是運行時斷言。例如，有限的浮點精度可能會導致違反這些承諾。
• 如果 is_X == False ，調用者應該期望操作符沒有 X 。
• 如果is_X == None(默認)，調用者應該沒有任何期望。