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


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


返回两个 SparseTensor 的元素最大值。

用法

tf.sparse.maximum(
    sp_a, sp_b, name=None
)

参数

  • sp_a 一个 SparseTensor 操作数,其 dtype 为实数,索引按字典顺序排列。
  • sp_b 另一个具有相同要求(和相同形状)的 SparseTensor 操作数。
  • name 操作的可选名称。

返回

  • output 输出稀疏张量。

假设两个 SparseTensor 具有相同的形状,即没有广播。

例子:

sp_zero = tf.sparse.SparseTensor([[0]], [0], [7])
sp_one = tf.sparse.SparseTensor([[1]], [1], [7])
res = tf.sparse.maximum(sp_zero, sp_one)
res.indices
<tf.Tensor:shape=(2, 1), dtype=int64, numpy=
array([[0],
       [1]])>
res.values
<tf.Tensor:shape=(2,), dtype=int32, numpy=array([0, 1], dtype=int32)>
res.dense_shape
<tf.Tensor:shape=(1,), dtype=int64, numpy=array([7])>

此元素操作的缩减版本是tf.sparse.reduce_max

相关用法


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