keras官方文档很简陋,只介绍了关键用法。那么来看看如何给自定义层传递其他参数。
新增一个参数名叫arg。
在自定义层的__init__函数中加入变量arg,并且赋值给成员变量:
调用时候可以直接传递:
想在自定义层中(我的是Attention层)使用,直接以成员变量的方式使用即可:
keras官方文档很简陋,只介绍了关键用法。那么来看看如何给自定义层传递其他参数。
新增一个参数名叫arg。
在自定义层的__init__函数中加入变量arg,并且赋值给成员变量:
调用时候可以直接传递:
想在自定义层中(我的是Attention层)使用,直接以成员变量的方式使用即可: