tf.contrib.distributions.tridiag
Создаёт матрицу со значениями выше, ниже и на диагонали.
tf.contrib.distributions.tridiag(
below=None, diag=None, above=None, name=None
)
Пример:
tridiag(below=[1., 2., 3.],
diag=[4., 5., 6., 7.],
above=[8., 9., 10.])
# ==> array([[ 4., 8., 0., 0.],
# [ 1., 5., 9., 0.],
# [ 0., 2., 6., 10.],
# [ 0., 0., 3., 7.]], dtype=float32)
| Аргументы |
|---|
below | Tensor формы [B1, ..., Bb, d-1], соответствующая части ниже диагонали. None логически эквивалентно below = 0. |
diag | Tensor формы [B1, ..., Bb, d], соответствующая диагональной части. None логически эквивалентно diag = 0. |
above | Tensor формы [B1, ..., Bb, d-1], соответствующая части выше диагонали. None логически эквивалентно above = 0. |
name | Python str. Имя для этого оператора. |
| Возвращаемое значение |
|---|
tridiag | Tensor со значениями, установленными выше, ниже и на диагонали. |
| Исключения |
|---|
ValueError | если все входные данные None. |
