當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。