静态图模式下如何将numpy数组传入层以作为常量使用?
收藏
0
收藏
全部评论(4)
可以试下 fill_constant
你好,fill_constant只能传入一个都是同样数字的数组,但是我想传一个我自己的numpy数组,这种情况应该怎么办呢?
试试这个呢?
param_name = fluid.layers.assign(np.array([1,2,3,4,5], dtype=np.int32))
0
请登录后评论
Hi,我在使用静态图模式时,需要向主程序传入一个常量以供使用,在静态图模式下如何实现传入一个常量呢?
我试过使用Tensor中的set()方法,但是在进行矩阵乘法的时候报错:
我只是想传入一个数组常量以和算出来的loss进行乘法计算