Add num_heads to attention node

test/onnx/attention_test.onnx

-attention_test:ä

-B
+attention_test:ö

+T
 input
 weights
 bias
 
-mask_indexresultAttention_0"	Attentionattention_testZ
+mask_indexresultAttention_0"	Attention*
+	num_heads 
attention_testZ
 input
 

 

test/onnx/gen_onnx.py

@@ -203,6 +203,8 @@ def attention_test():
     node = helper.make_node('Attention',
                             inputs=['input', 'weights', 'bias', 'mask_index'],
                             outputs=['result'],
+                            num_heads=12,
+                            
                             name="Attention_0")
 
     return ([node], [input, weights, bias, mask_index], [result])