首页 Paddle框架 帖子详情
【求助】paddle中如何实现tensorflow中自定义层的build功能
收藏
快速回复
Paddle框架 问答深度学习 342 3
【求助】paddle中如何实现tensorflow中自定义层的build功能
收藏
快速回复
Paddle框架 问答深度学习 342 3

如题目,

想问下,就是在paddle里面,我如何在init阶段还不知道特征尺寸的情况下,初始化一些参数矩阵,

我可以在forward里面定义训练参数不?不在init里定义,而是在forward里,因为我在init时,还不能确定参数矩阵的size,

这个功能在tensorflow里面可以通过在init和call中间,自己重写build(input_shape)方法实现这个操作

0
收藏
回复
全部评论(3)
时间顺序
JavaRoom
#2 回复于2022-01

你也重写试试,用个动态的列表啥的,看行不行?

1
回复
橘枳
#3 回复于2022-01
你也重写试试,用个动态的列表啥的,看行不行?

功力还没到这一层

0
回复
JavaRoom
#4 回复于2022-01
橘枳 #3
功力还没到这一层

举个例子,比如resnet,添加多少层,都是动态的,根据参数来的啊。

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户