标签: tensorflow
当然,我知道如何使用numpy等初始化我的特定项目使用的权重,但我想知道如何将init策略添加到张量流,这是我想为项目做出贡献。哪个文件有像Xavier这样的init策略的实现,以便我可以将He等添加到它? 此外,如果我想将l-BFGS添加到包中,我甚至可以在哪里开始寻找?
任何指针都非常感谢!
答案 0 :(得分:1)
您可以在此处找到Xavier初始化的实现。
https://github.com/tensorflow/tensorflow/blob/6b1d4fd8090d44d20fdadabf06f1a9b178c3d80c/tensorflow/contrib/layers/python/layers/initializers.py#L31