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


Python tf.sparse.SparseTensor.with_values用法及代碼示例


用法

with_values(
    new_values
)

參數

  • new_values SparseTensor 的值。需要與當前的 .values Tensor 具有相同的形狀。可能具有與當前 values 不同的類型。

返回

  • SparseTensor 具有相同的索引和形狀,但更新了值。

返回 self 的副本,其中 values 替換為 new_values

此方法生成一個新的 SparseTensor,它具有相同的非零 indices 和相同的 dense_shape ,但更新了值。

示例用法:

st = tf.sparse.from_dense([[1, 0, 2, 0], [3, 0, 0, 4]])
tf.sparse.to_dense(st.with_values([10, 20, 30, 40]))  # 4 nonzero values
<tf.Tensor:shape=(2, 4), dtype=int32, numpy=
array([[10,  0, 20,  0],
       [30,  0,  0, 40]], dtype=int32)>

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.sparse.SparseTensor.with_values。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。