pytorch2.5.1:é
Œ


x
cos
sinoutput/Custom"Custom*
input_namesJ
xJcosJsin 
*
mode
 
*
output_namesJ
y 
*"
type"RotaryPositionEmbedding 

main_graphZ

x




€ˆ


€
Z 
cos




€ˆ



€
Z 
sin




€ˆ



€
bh
output^
\
X
Customoutput_dim_0
Customoutput_dim_1
Customoutput_dim_2
Customoutput_dim_3B