提交 a437d052 编写于 作者: W wjj19950828

add notes

上级 8614aac8
...@@ -2586,6 +2586,7 @@ class OpSet9(): ...@@ -2586,6 +2586,7 @@ class OpSet9():
k = _const_weight_or_none(val_k) k = _const_weight_or_none(val_k)
if isinstance(k, (list, tuple, np.ndarray)): if isinstance(k, (list, tuple, np.ndarray)):
k = k[0] k = k[0]
# If k can get the value directly, it is used as an attribute; otherwise it is used as an input tensor
if k is not None: if k is not None:
layer_attrs["k"] = k layer_attrs["k"] = k
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册