tf.linalg.LinearOperatorCirculant
LinearOperator действующий как циркулянтная матрица.
Наследуется от: LinearOperator, Module
tf.linalg.LinearOperatorCirculant(
spectrum: tf.Tensor,
input_output_dtype=tf.dtypes.complex64,
is_non_singular: bool = None,
is_self_adjoint: bool = None,
is_positive_definite: bool = None,
is_square: bool = True,
name='LinearOperatorCirculant'
)
Этот оператор действует как циркулянтная матрица A с формой [B1,...,Bb, N, N] для некоторого b >= 0. Первые b индексы индексируют элемент пакета. Для каждого индекса пакета (i1,...,ib), A[i1,...,ib, : :] является N x N матрицей. Эта матрица A не материализуется, но для целей вещания эта форма будет актуальна.
Описание в терминах циркулянтных матриц
Циркулянт означает, что элементы A генерируются одним вектором, ядром свёртки h: A_{mn} := h_{m-n mod N}. При h = [w, x, y, z],
A = |w z y x|
|x w z y|
|y x w z|
|z y x w|
Это означает, что результат матричного умножения v = Au имеет Lth столбец, заданный циклической свёрткой между h и Lth столбцом u.
Описание в терминах спектра частот
Существует эквивалентное описание в терминах [пакетного] спектра H и преобразований Фурье. Здесь мы рассматриваем A.shape = [N, N] и игнорируем размерности пакетов. Определим дискретное преобразование Фурье (ДПФ) и его обратное:
DFT[ h[n] ] = H[k] := sum_{n = 0}^{N - 1} h_n e^{-i 2pi k n / N}
IDFT[ H[k] ] = h[n] = N^{-1} sum_{k = 0}^{N - 1} H_k e^{i 2pi k n / N}
Из этих определений мы видим, что
H[0] = sum_{n = 0}^{N - 1} h_n
H[1] = "the first positive frequency"
H[N - 1] = "the first negative frequency"
В широком смысле, при * поэлементном умножении матричное умножение равно действию множителя Фурье: A u = IDFT[ H * DFT[u] ]. Точнее, заданная [N, R] матрица u, пусть DFT[u] — [N, R] матрица со rth столбцом, равным ДПФ rth столбца u. Определим IDFT аналогично. Матричное умножение можно выразить по столбцам:
(A u)_r = IDFT[ H * (DFT[u])_r ]
Свойства оператора, выведенные из спектра.
Пусть U — kth евклидов базисный вектор, и U = IDFT[u]. Вышеприведенные формулы показывают, чтоA U = H_k * U. Мы делаем вывод, что элементы H являются собственными значениями этого оператора. Следовательно
- Этот оператор является положительно определённым тогда и только тогда, когда
Real{H} > 0.
Общим свойством преобразований Фурье является соответствие между эрмитовыми функциями и вещественными преобразованиями.
Предположим H.shape = [B1,...,Bb, N]. Мы говорим, что H является эрмитовым спектром, если при % понимается как модульное деление,
H[..., n % N] = ComplexConjugate[ H[..., (-n) % N] ]
- Этот оператор соответствует вещественной матрице тогда и только тогда, когда
Hявляется эрмитовой. - Этот оператор является самосопряжённым тогда и только тогда, когда
Hвещественное.
См., например, "Цифровые сигналы", Оппенхейм и Шафер.
Пример самосопряжённого положительно определённого оператора
# spectrum is real ==> operator is self-adjoint
# spectrum is positive ==> operator is positive definite
spectrum = [6., 4, 2]
operator = LinearOperatorCirculant(spectrum)
# IFFT[spectrum]
operator.convolution_kernel()
==> [4 + 0j, 1 + 0.58j, 1 - 0.58j]
operator.to_dense()
==> [[4 + 0.0j, 1 - 0.6j, 1 + 0.6j],
[1 + 0.6j, 4 + 0.0j, 1 - 0.6j],
[1 - 0.6j, 1 + 0.6j, 4 + 0.0j]]
Пример определения в терминах вещественного ядра свёртки
# convolution_kernel is real ==> spectrum is Hermitian.
convolution_kernel = [1., 2., 1.]]
spectrum = tf.signal.fft(tf.cast(convolution_kernel, tf.complex64))
# spectrum is Hermitian ==> operator is real.
# spectrum is shape [3] ==> operator is shape [3, 3]
# We force the input/output type to be real, which allows this to operate
# like a real matrix.
operator = LinearOperatorCirculant(spectrum, input_output_dtype=tf.float32)
operator.to_dense()
==> [[ 1, 1, 2],
[ 2, 1, 1],
[ 1, 2, 1]]
Пример эрмитова спектра
# spectrum is shape [3] ==> operator is shape [3, 3]
# spectrum is Hermitian ==> operator is real.
spectrum = [1, 1j, -1j]
operator = LinearOperatorCirculant(spectrum)
operator.to_dense()
==> [[ 0.33 + 0j, 0.91 + 0j, -0.24 + 0j],
[-0.24 + 0j, 0.33 + 0j, 0.91 + 0j],
[ 0.91 + 0j, -0.24 + 0j, 0.33 + 0j]
Пример принудительного вещественного dtype при эрмитовом спектре
# spectrum is shape [4] ==> operator is shape [4, 4]
# spectrum is real ==> operator is self-adjoint
# spectrum is Hermitian ==> operator is real
# spectrum has positive real part ==> operator is positive-definite.
spectrum = [6., 4, 2, 4]
# Force the input dtype to be float32.
# Cast the output to float32. This is fine because the operator will be
# real due to Hermitian spectrum.
operator = LinearOperatorCirculant(spectrum, input_output_dtype=tf.float32)
operator.shape
==> [4, 4]
operator.to_dense()
==> [[4, 1, 0, 1],
[1, 4, 1, 0],
[0, 1, 4, 1],
[1, 0, 1, 4]]
# convolution_kernel = tf.signal.ifft(spectrum)
operator.convolution_kernel()
==> [4, 1, 0, 1]
Производительность
Предположим, operator — это LinearOperatorCirculant с формой [N, N], а x.shape = [N, R]. Тогда
-
operator.matmul(x)являетсяO(R*N*Log[N]) -
operator.solve(x)являетсяO(R*N*Log[N]) -
operator.determinant()включает в себяNreduce_prodразмера.
Если вместо этого operator и x имеют форму [B1,...,Bb, N, N] и [B1,...,Bb, N, R], каждая операция увеличивает сложность на B1*...*Bb.
Подсказки о свойствах матрицы
Этот LinearOperator инициализируется флагами типа is_X для X = non_singular, self_adjoint, positive_definite, square. Они имеют следующее значение:
- Если
is_X == True, вызывающие стороны должны ожидать, что оператор будет обладать свойствомX. Это обещание, которое должно выполняться, но это не проверка во время выполнения. Например, конечная точность с плавающей запятой может привести к нарушению этих обещаний. - Если
is_X == False, вызывающие стороны должны ожидать, что оператор не будет обладать свойствомX. - Если
is_X == None(по умолчанию), вызывающие стороны не должны ожидать ничего определённого.
| Ссылки | |
|---|---|
| Обзор матриц Топлица и Циркулянт: Грей, 2006 (pdf) |
| Аргументы | |
|---|---|
spectrum | Форма [B1,...,Bb, N] Tensor. Разрешенные типы данных: float16, float32, float64, complex64, complex128. Тип может отличаться от input_output_dtype |
input_output_dtype | dtype для ввода/вывода. |
is_non_singular | Ожидается, что этот оператор невырожден. |
is_self_adjoint | Ожидается, что этот оператор равен своему эрмитовому транспонированию. Если spectrum вещественный, это всегда будет истинно. |
is_positive_definite | Ожидается, что этот оператор является положительно определённым, то есть квадратичная форма x^H A x имеет положительную вещественную часть для всех ненулевых x. Обратите внимание, что мы не требуем, чтобы оператор был самосопряжённым, чтобы быть положительно определённым. См.: https://en.wikipedia.org/wiki/Positive-definite_matrix#Extension_for_non_symmetric_matrices |
is_square | Ожидается, что этот оператор действует как квадратные [пакетные] матрицы. |
name | Имя для добавления ко всем операциям, созданным этим классом. |
| Атрибуты | |
|---|---|
H | Возвращает сопряжённую матрицу текущего LinearOperator. Учитывая |
batch_shape | Размерность пакетных измерений этого LinearOperator. Если этот оператор действует как пакетная матрица |
block_depth | Глубина рекурсивно определённых циркулянтных блоков, определяющих этот Operator. С
A = |w z y x|
|x w z y|
|y x w z|
|z y x w|
A = |W Z Y X|
|X W Z Y|
|Y X W Z|
|Z Y X W|
|
block_shape | |
domain_dimension | Размерность (в смысле векторных пространств) области действия этого оператора. Если этот оператор действует как пакетная матрица |
dtype | DType обрабатываемых этим LinearOperator. |
graph_parents | Список графовых зависимостей этого LinearOperator. (устарело)
|
is_non_singular | |
is_positive_definite | |
is_self_adjoint | |
is_square | Возвращает True/False в зависимости от того, является ли этот оператор квадратным. |
parameters | Словарь параметров, используемых для создания этого LinearOperator. |
range_dimension | Размерность (в смысле векторных пространств) области значений этого оператора. Если этот оператор действует как пакетная матрица |
shape | Размерность этого LinearOperator. Если этот оператор действует как пакетная матрица |
spectrum | |
tensor_rank | Ранг (в смысле тензоров) матрицы, соответствующей этому оператору. Если этот оператор действует как пакетная матрица |
Методы
add_to_tensor
add_to_tensor(
x, name='add_to_tensor'
)
Добавляет матрицу, представленную этим оператором, к x. Эквивалентно A + x.
| Аргументы | |
|---|---|
x | Tensor с той же размерностью и формой, совместимой для вещания с self.shape. |
name | Имя, которое нужно дать этому Op. |
| Возвращаемое значение | |
|---|---|
Tensor с формой, полученной в результате вещания, и той же размерностью, что и у self. |
adjoint
adjoint(
name: str = 'adjoint'
) -> 'LinearOperator'
Возвращает сопряжённую матрицу текущего LinearOperator.
Учитывая A, представляющее этот LinearOperator, возвращает A*. Обратите внимание, что вызов self.adjoint() и self.H эквивалентны.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
LinearOperator, представляющий сопряжённую матрицу этого LinearOperator. |
assert_hermitian_spectrum
assert_hermitian_spectrum(
name='assert_hermitian_spectrum'
)
Возвращает утверждение, что спектр этого оператора эрмитов.
Этот оператор соответствует вещественной матрице тогда и только тогда, когда её спектр эрмитов.
| Аргументы | |
|---|---|
name | Имя для этого утверждения. |
| Возвращаемое значение | |
|---|---|
| Утверждение, что спектр этого оператора эрмитов. |
assert_non_singular
assert_non_singular(
name='assert_non_singular'
)
Возвращает утверждение, что этот оператор невырожден.
Этот оператор считается невырожденным, если
ConditionNumber < max{100, range_dimension, domain_dimension} * eps,
eps := np.finfo(self.dtype.as_numpy_dtype).eps
| Аргументы | |
|---|---|
name | Имя строки, добавляемое к создаваемым операциям. |
| Возвращаемое значение | |
|---|---|
Объект Assert Op, который при выполнении вызовет исключение InvalidArgumentError, если оператор вырожден. |
assert_positive_definite
assert_positive_definite(
name='assert_positive_definite'
)
Возвращает объект, проверяющий, что этот оператор является положительно определённым.
Здесь положительно определённым означает, что квадратичная форма x^H A x имеет положительную действительную часть для всех ненулевых x. Обратите внимание, что для положительной определённости оператор не обязан быть самосопряжённым.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
Объект Assert Op, который при выполнении вызовет исключение InvalidArgumentError, если оператор не является положительно определённым. |
assert_self_adjoint
assert_self_adjoint(
name='assert_self_adjoint'
)
Возвращает объект, проверяющий, что этот оператор является самосопряжённым.
Здесь проверяется, что этот оператор точно равен своему эрмитовому транспонированию.
| Аргументы | |
|---|---|
name | Имя строки, добавляемое к создаваемым операциям. |
| Возвращаемое значение | |
|---|---|
Объект Assert Op, который при выполнении вызовет исключение InvalidArgumentError, если оператор не является самосопряжённым. |
batch_shape_tensor
batch_shape_tensor(
name='batch_shape_tensor'
)
Форма размерностей пакета этого оператора, определяемая во время выполнения.
Если этот оператор действует как пакетная матрица A с A.shape = [B1,...,Bb, M, N], то возвращает объект Tensor, содержащий [B1,...,Bb].
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
int32 Tensor |
block_shape_tensor
block_shape_tensor()
Форма блочных размеров self.spectrum.
cholesky
cholesky(
name: str = 'cholesky'
) -> 'LinearOperator'
Возвращает фактор Холецкого как LinearOperator.
Учитывая A, представляющий этот LinearOperator, если A является положительно определенной самосопряжённой матрицей, возвращает L, где A = L L^T, т. е. разложение Холецкого.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
LinearOperator, представляющий нижнюю треугольную матрицу в разложении Холецкого. |
| Исключения | |
|---|---|
ValueError | Если LinearOperator не указано, что он является положительно определённой и самосопряжённой матрицей. |
cond
cond(
name='cond'
)
Возвращает число обусловленности этого линейного оператора.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
Матрица формы [B1,...,Bb] Tensor одинаковой формы dtype как self. |
convolution_kernel
convolution_kernel(
name='convolution_kernel'
)
Ядро свёртки, соответствующее self.spectrum.
D-мерное преобразование Фурье этого ядра является спектром этого оператора в частотной области.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
Tensor с dtype self.dtype. |
determinant
determinant(
name='det'
)
Определитель для каждого элемента набора.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
Tensor с формой self.batch_shape и одинаковой формой dtype как self. |
| Исключения | |
|---|---|
NotImplementedError | Если self.is_square является False. |
diag_part
diag_part(
name='diag_part'
)
Эффективный способ получить [пакетный] диагональный элемент этого оператора.
Если у этого оператора есть форма [B1,...,Bb, M, N], это возвращает Tensor diagonal, с формой [B1,...,Bb, min(M, N)], где diagonal[b1,...,bb, i] = self.to_dense()[b1,...,bb, i, i].
my_operator = LinearOperatorDiag([1., 2.]) # Efficiently get the diagonal my_operator.diag_part() ==> [1., 2.] # Equivalent, but inefficient method tf.linalg.diag_part(my_operator.to_dense()) ==> [1., 2.]
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращает | |
|---|---|
diag_part | Tensor того же dtype, что и self. |
domain_dimension_tensor
domain_dimension_tensor(
name='domain_dimension_tensor'
)
Размерность (в смысле векторных пространств) области действия этого оператора.
Определяется во время выполнения.
Если этот оператор действует как матрица A с A.shape = [B1,...,Bb, M, N], то это возвращает N.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращает | |
|---|---|
int32 Tensor |
eigvals
eigvals(
name='eigvals'
)
Возвращает собственные значения этого линейного оператора.
Если оператор помечен как самосопряженный (через is_self_adjoint), это вычисление может быть более эффективным.
Примечание: В настоящее время поддерживаются только самосопряженные операторы.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращает | |
|---|---|
Форма [B1,...,Bb, N] Tensor того же dtype, что и self. |
inverse
inverse(
name: str = 'inverse'
) -> 'LinearOperator'
Возвращает обратный LinearOperator.
Дано A, представляющее этот LinearOperator, вернуть LinearOperator, представляющий A^-1.
| Аргументы | |
|---|---|
name | Диапазон имен для использования для операций, добавленных этим методом. |
| Возвращает | |
|---|---|
LinearOperator, представляющий обратную матрицу. |
| Возможные исключения | |
|---|---|
ValueError | Если LinearOperator не указано как non_singular. |
log_abs_determinant
log_abs_determinant(
name='log_abs_det'
)
Логарифм абсолютного значения определителя для каждого элемента пакета.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращает | |
|---|---|
Tensor с формой self.batch_shape и тем же dtype, что и self. |
| Возможные исключения | |
|---|---|
NotImplementedError | Если self.is_square является False. |
matmul
matmul(
x, adjoint=False, adjoint_arg=False, name='matmul'
)
Преобразовать матрицу [пакет] x с левым умножением: x --> Ax.
# Make an operator acting like batch matrix A. Assume A.shape = [..., M, N] operator = LinearOperator(...) operator.shape = [..., M, N] X = ... # shape [..., N, R], batch matrix, R > 0. Y = operator.matmul(X) Y.shape ==> [..., M, R] Y[..., :, r] = sum_j A[..., :, j] X[j, r]
| Аргументы | |
|---|---|
x | LinearOperator или Tensor с совместимой формой и тем же dtype, что и у self. См. строку документации класса для определения совместимости. |
adjoint | Питоновская bool. Если True, умножайте слева на сопряжённое: A^H x. |
adjoint_arg | Питоновская bool. Если True, вычисляйте A x^H, где x^H — это эрмитово сопряжение (транспонирование и комплексное сопряжение). |
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
A LinearOperator или Tensor с формой [..., M, R] и тем же dtype, что и у self. |
matvec
matvec(
x, adjoint=False, name='matvec'
)
Преобразовать [пакетный] вектор x с помощью левого умножения: x --> Ax.
# Make an operator acting like batch matrix A. Assume A.shape = [..., M, N] operator = LinearOperator(...) X = ... # shape [..., N], batch vector Y = operator.matvec(X) Y.shape ==> [..., M] Y[..., :] = sum_j A[..., :, j] X[..., j]
| Аргументы | |
|---|---|
x | Tensor с совместимой формой и тем же dtype, что и у self. x рассматривается как [пакетный] вектор, то есть для каждого набора ведущих измерений последнее измерение определяет вектор. См. строку документации класса для определения совместимости. |
adjoint | Питоновская bool. Если True, умножайте слева на сопряжённое: A^H x. |
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
A Tensor с формой [..., M] и тем же dtype, что и у self. |
range_dimension_tensor
range_dimension_tensor(
name='range_dimension_tensor'
)
Размерность (в смысле векторных пространств) области значений этого оператора.
Определяется во время выполнения.
Если этот оператор действует как пакетная матрица A с A.shape = [B1,...,Bb, M, N], то он возвращает M.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
int32 Tensor |
shape_tensor
shape_tensor(
name='shape_tensor'
)
Форма этого LinearOperator, определяется во время выполнения.
Если этот оператор действует как пакетная матрица A с A.shape = [B1,...,Bb, M, N], то он возвращает Tensor, содержащий [B1,...,Bb, M, N], эквивалентное tf.shape(A).
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
int32 Tensor |
solve
solve(
rhs, adjoint=False, adjoint_arg=False, name='solve'
)
Решить (точно или приближённо) R (пакетные) системы уравнений: A X = rhs.
Возвращаемое Tensor будет близко к точному решению, если A хорошо обусловлен. В противном случае близость будет варьироваться. См. строку документации класса для получения подробной информации.
Примеры:
# Make an operator acting like batch matrix A. Assume A.shape = [..., M, N] operator = LinearOperator(...) operator.shape = [..., M, N] # Solve R > 0 linear systems for every member of the batch. RHS = ... # shape [..., M, R] X = operator.solve(RHS) # X[..., :, r] is the solution to the r'th linear system # sum_j A[..., :, j] X[..., j, r] = RHS[..., :, r] operator.matmul(X) ==> RHS
| Аргументы | |
|---|---|
rhs | Tensor с той же dtype, что и у этого оператора, и совместимой формой. rhs рассматривается как [пакетная] матрица, то есть для каждого набора ведущих измерений последние два измерения определяют матрицу. См. строку документации класса для определения совместимости. |
adjoint | Питоновская bool. Если True, решить систему, включающую сопряжённый оператор этого LinearOperator: A^H X = rhs. |
adjoint_arg | Питоновская bool. Если True, решить A X = rhs^H, где rhs^H — это эрмитово сопряжение (транспонирование и комплексное сопряжение). |
name | Область имён, которую следует использовать для операций, добавленных этим методом. |
| Возвращаемое значение | |
|---|---|
Tensor с формой [...,N, R] и тем же dtype, что и у rhs. |
| Исключения | |
|---|---|
NotImplementedError | Если self.is_non_singular или is_square равно False. |
solvevec
solvevec(
rhs, adjoint=False, name='solve'
)
Решить одно уравнение с максимальной точностью: A X = rhs.
Возвращаемое Tensor будет близко к точному решению, если A хорошо обусловлен. В противном случае близость будет варьироваться. См. строку документации класса для получения подробной информации.
Примеры:
# Make an operator acting like batch matrix A. Assume A.shape = [..., M, N] operator = LinearOperator(...) operator.shape = [..., M, N] # Solve one linear system for every member of the batch. RHS = ... # shape [..., M] X = operator.solvevec(RHS) # X is the solution to the linear system # sum_j A[..., :, j] X[..., j] = RHS[..., :] operator.matvec(X) ==> RHS
| Аргументы | |
|---|---|
rhs | Tensor с той же dtype, что и у этого оператора. rhs рассматривается как [пакетный] вектор, то есть для каждого набора ведущих измерений последнее измерение определяет вектор. См. строку документации класса для определения совместимости по пакетным измерениям. |
adjoint | Питоновская bool. Если True, решить систему, включающую сопряжённый оператор этого LinearOperator: A^H X = rhs. |
name | Область имён, которую следует использовать для операций, добавленных этим методом. |
| Возвращаемое значение | |
|---|---|
Tensor с формой [...,N] и тем же dtype, что и у rhs. |
| Исключения | |
|---|---|
NotImplementedError | Если self.is_non_singular или is_square равно False. |
tensor_rank_tensor
tensor_rank_tensor(
name='tensor_rank_tensor'
)
Ранг (в смысле тензоров) матрицы, соответствующей этому оператору.
Если этот оператор действует как пакетная матрица A с A.shape = [B1,...,Bb, M, N], то он возвращает b + 2.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
int32 Tensor, определяется во время выполнения. |
to_dense
to_dense(
name='to_dense'
)
Вернуть плотную (пакетную) матрицу, представляющую этот оператор.
trace
trace(
name='trace'
)
След линейного оператора, равный сумме self.diag_part().
Если оператор квадратный, это также сумма собственных значений.
| Аргументы | |
|---|---|
name | Имя для этого Op. |
| Возвращаемое значение | |
|---|---|
Фигура [B1,...,Bb] Tensor того же dtype, что и self. |
__getitem__
__getitem__(
slices
)
__matmul__
__matmul__(
other
)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/linalg/LinearOperatorCirculant