tf.contrib.kfac.utils.matmul_sparse_dense

tf.contrib.kfac.utils.matmul_sparse_dense(
    A,
    B,
    name=None
)

Defined in tensorflow/contrib/kfac/python/ops/utils.py.

Computes matmul(A, B) where A is sparse, B is dense.

Args:

  • A: tf.IndexedSlices with dense shape [m, n].
  • B: tf.Tensor with shape [n, k].
  • name: str. Name of op.

Returns:

tf.IndexedSlices resulting from matmul(A, B).

Raises:

  • ValueError: If A doesn't represent a matrix.
  • ValueError: If B is not rank-2.