tf.contrib.distributions.tridiag

View source on GitHub

Creates a matrix with values set above, below, and on the diagonal.

tf.contrib.distributions.tridiag(
    below=None, diag=None, above=None, name=None
)

Example:

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 of shape [B1, ..., Bb, d-1] corresponding to the below diagonal part. None is logically equivalent to below = 0.
diag Tensor of shape [B1, ..., Bb, d] corresponding to the diagonal part. None is logically equivalent to diag = 0.
above Tensor of shape [B1, ..., Bb, d-1] corresponding to the above diagonal part. None is logically equivalent to above = 0.
name Python str. The name to give this op.

tridiag Tensor with values set above, below and on the diagonal.

ValueError if all inputs are None.