【发布时间】: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
我想构造一个权重,它的某些元素为零且永不改变,其他元素为变量。例如:
[[0,0,a,0],[0,0,b,0],[0,0,0,c],[0,0,0,d]]
这是一个 tf 变量,所有零都保持不变。只有 a、b、c、d 使用梯度下降进行了调整。
有谁知道如何定义这样的矩阵?
【问题讨论】:
标签: tensorflow
您应该查看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])
【讨论】: