Julia Flux，稠密初始条件

Question

我想不出初始化Flux.Dense.权重的语法

using Flux 
w0 = Float32[1, 2, 3, 4] 
nn = Dense(4, 1, relu; initW = w0) 

会产生以下错误。

MethodError: objects of type Array{Float32,2} are not callable

看来initW需要是某种函数。

我也试过

nn = Dense(4, 1, relu)
nn.W = w0

并得到错误

setfield! immutable struct of type Dense cannot be changed

这似乎是在说我无法更改稠密对象(？(。

谢谢你的光临！

(Julia 1.4.2和Flux v0.11.0(

Answer 1

initW参数不是权重向量，它用于更改初始化权重的函数。你可以放一个函数或尝试

nn = Dense(4, 1, relu)
nn.W .= w0

操作员。=更新值。

不幸的是，我无法访问我的电脑进行检查。