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.