【问题标题】:How to construct a tensor weights whose certain elements are zero?如何构造某些元素为零的张量权重?
【发布时间】:2017-12-08 16:44:40
【问题描述】:

我想构造一个权重,它的某些元素为零且永不改变,其他元素为变量。例如:

[[0,0,a,0],[0,0,b,0],[0,0,0,c],[0,0,0,d]]

这是一个 tf 变量,所有零都保持不变。只有 a、b、c、d 使用梯度下降进行了调整。

有谁知道如何定义这样的矩阵?

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    您应该查看SparseTensor。它针对张量由许多零组成的操作进行了高度优化。

    所以,在你的情况下,初始化 SparseTensor:

    a,b,c,d = 10,20,30,40
    sparse = tf.SparseTensor([[0,2], [1,2], [2,3], [3,3]], [a,b,c,d], [4,4])
    

    【讨论】:

    • 然而,我需要的是一个稀疏的 tf.Variable。现在看来这是一个常数张量?
    • This question 看起来很相关。 TensorFlow github 上还有 this feature request
    猜你喜欢
    • 2018-12-31
    • 2022-08-16
    • 2021-08-30
    • 1970-01-01
    • 2016-09-22
    • 2020-01-27
    • 2019-11-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多