当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.sparse.from_dense用法及代码示例


将密集张量转换为稀疏张量。

用法

tf.sparse.from_dense(
    tensor, name=None
)

参数

  • tensor 将要转换为 SparseTensor 的密集 Tensor
  • name 操作的可选名称。

返回

  • SparseTensor

结果中只会出现不等于 0 的元素。生成的 SparseTensor 具有与输入相同的 dtype 和 shape。

sp = tf.sparse.from_dense([0, 0, 3, 0, 1])
sp.shape.as_list()
[5]
sp.values.numpy()
array([3, 1], dtype=int32)
sp.indices.numpy()
array([[2],
       [4]])

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.sparse.from_dense。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。