TensorFlow函數(shù)：tf.sparse_matmul

tf.sparse_matmul 函數(shù)

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

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

輸入必須是二維矩陣,“a”的內(nèi)部維度必須與“b”的外部維度相匹配.該操作針對“a”或“b”中至少一個稀疏的情況進(jìn)行了優(yōu)化.在一個平臺上使用這個與稠密矩陣相乘的平衡點(diǎn)在稀疏矩陣中為30％零值.

這個操作的梯度計算只會利用梯度來自 Relu 時輸入梯度的稀疏性.

函數(shù)參數(shù)：

• a：A Tensor,必須是以下類型之一：float32,bfloat16.
• b：A Tensor,必須是以下類型之一：float32,bfloat16.
• transpose_a：可選的bool,默認(rèn)為False.
• transpose_b：可選的bool,默認(rèn)為False.
• a_is_sparse：可選的bool,默認(rèn)為False.
• b_is_sparse：可選的bool,默認(rèn)為False.
• name：操作的名稱(可選).

函數(shù)返回值：

該 tf.sparse_matmul 函數(shù)返回一個float32類型的Tensor.