如何在张量流中对变量进行复杂的初始化?

时间:2016-06-28 07:17:30

标签: tensorflow

我有一个变量,它将成为CNN的权重。让我们说它的形状[kernel_width,kernel_height,in_channels,out_channels]。

我想要做的是用一些随机值初始化它,但每个内核的中心值为1。

例如,对于3x3内核,我希望值为[[r,r,r],[r,1,r],[r,r,r]],其中每个r是不同的随机数从任何分布中获取的价值。

我可以轻松地以该格式创建一个numpy数组,但我看不出如何使用任意数组来初始化变量。

1 个答案:

答案 0 :(得分:1)

正如您所说,最简单的方法是使用numpy数组初始化过滤器。您只需在创建变量时传递此数组:

grid.masonry('reloadItems');
grid.masonry('layout');