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


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。