W3cubDocs

/TensorFlow Python

tf.contrib.distributions.tridiag

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

Defined in tensorflow/python/ops/distributions/util.py.

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

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)

Args:

  • 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.

Returns:

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

Raises:

  • ValueError: if all inputs are None.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/distributions/tridiag