小编Mar*_*ski的帖子

Tensorflow 自定义层:使用可训练参数创建稀疏矩阵

我正在研究的模型应该同时预测相当多的变量(>1000)。因此,我希望在每个输出的网络末端都有一个小型神经网络。

为了紧凑地做到这一点,我想找到一种方法在 Tensorflow 框架内的神经网络的两层之间创建稀疏的可训练连接。

连接矩阵中只有一小部分应该是可训练的:只有参数是块对角线的一部分。


例如: 参见不密集部分

连接矩阵如下:

分块对角矩阵

可训练参数应该位于1的位置。

python machine-learning tensorflow

6
推荐指数
1
解决办法
2061
查看次数

标签 统计

machine-learning ×1

python ×1

tensorflow ×1