TensorFlow函数:tf.sparse_matmul

由 Carrie 创建, 最后一次修改 2018-02-27

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.

以上内容是否对您有帮助:
二维码
建议反馈
二维码