tf.sparse_matmul 函数

sparse_matmul(
    a,
    b,
    transpose_a=False,
    transpose_b=False,
    a_is_sparse=False,
    b_is_sparse=False,
    name=None
)

用矩阵“b”乘以矩阵“a”.

输入必须是二维矩阵,“a”的内部维度必须与“b”的外部维度相匹配.该操作针对“a”或“b”中至少一个稀疏的情况进行了优化.在一个平台上使用这个与稠密矩阵相乘的平衡点在稀疏矩阵中为30％零值.

这个操作的梯度计算只会利用梯度来自 Relu 时输入梯度的稀疏性.

函数参数：

• a：A Tensor,必须是以下类型之一：float32,bfloat16.
• b：A Tensor,必须是以下类型之一：float32,bfloat16.
• transpose_a：可选的bool,默认为False.
• transpose_b：可选的bool,默认为False.
• a_is_sparse：可选的bool,默认为False.
• b_is_sparse：可选的bool,默认为False.
• name：操作的名称(可选).

函数返回值：

该 tf.sparse_matmul 函数返回一个float32类型的Tensor.