{"id":10020952,"date":"2025-08-19T21:25:37","date_gmt":"2025-08-19T13:25:37","guid":{"rendered":"https:\/\/itmax.cn\/?p=10020952"},"modified":"2025-08-19T21:25:37","modified_gmt":"2025-08-19T13:25:37","slug":"%e3%80%90triton-%e6%95%99%e7%a8%8b%e3%80%91triton_language-broadcast_to","status":"publish","type":"post","link":"https:\/\/itmax.cn\/?p=10020952","title":{"rendered":"\u3010Triton \u6559\u7a0b\u3011triton_language.broadcast_to"},"content":{"rendered":"<p>Triton \u662f\u4e00\u79cd\u7528\u4e8e\u5e76\u884c\u7f16\u7a0b\u7684\u8bed\u8a00\u548c\u7f16\u8bd1\u5668\u3002\u5b83\u65e8\u5728\u63d0\u4f9b\u4e00\u4e2a\u57fa\u4e8e Python \u7684\u7f16\u7a0b\u73af\u5883\uff0c\u4ee5\u9ad8\u6548\u7f16\u5199\u81ea\u5b9a\u4e49 DNN \u8ba1\u7b97\u5185\u6838\uff0c\u5e76\u80fd\u591f\u5728\u73b0\u4ee3 GPU \u786c\u4ef6\u4e0a\u4ee5\u6700\u5927\u541e\u5410\u91cf\u8fd0\u884c\u3002<\/p>\n<p>\u66f4\u591a Triton \u4e2d\u6587\u6587\u6863\u53ef\u8bbf\u95ee \u2192<a href=\"https:\/\/link.segmentfault.com\/?enc=%2F3VX35UgFUz1T5laXuVTvg%3D%3D.%2BH4A5ttw7H19vST9agtXcD4G1NaEM4MKCUbRd88X5Tg%3D\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">https:\/\/triton.hyper.ai\/<\/a><\/p>\n<pre><code>triton.language.broadcast_to(input, *shape)<\/code><\/pre>\n<p>\u5c1d\u8bd5\u5c06\u7ed9\u5b9a\u7684\u5f20\u91cf\u5e7f\u64ad\u5230\u65b0\u7684\u00a0<code>shape<\/code>\u3002<\/p>\n<p><strong>\u53c2\u6570<\/strong>\uff1a<\/p>\n<ul>\n<li>input (Block) &#8211; \u8f93\u5165\u5f20\u91cf\u3002<\/li>\n<li>shape &#8211; \u6240\u9700\u7684\u5f62\u72b6\u3002<\/li>\n<\/ul>\n<p><code>shape<\/code>\u00a0\u53ef\u4ee5\u4ee5 1 \u4e2a\u5143\u7ec4\u6216\u72ec\u7acb\u53c2\u6570\u88ab\u4f20\u5165\uff1a<\/p>\n<pre><code># These are equivalent \n# \u8fd9\u4e9b\u662f\u7b49\u6548\u7684\nbroadcast_to(x, (32, 32))\nbroadcast_to(x, 32, 32)<\/code><\/pre>\n<p>\u8be5\u51fd\u6570\u4e5f\u53ef\u4f5c\u4e3a\u00a0<code>tensor<\/code>\u00a0\u7684 1 \u4e2a\u6210\u5458\u51fd\u6570\u8c03\u7528\uff0c\u4f7f\u7528\u00a0<code>x.broadcast_to(...)<\/code>\u00a0\u800c\u4e0d\u662f\u00a0<code>broadcast_to(x, ...)<\/code>\u3002<\/p>\n<p><a href=\"https:\/\/segmentfault.com\/a\/1190000047148676\">https:\/\/segmentfault.com\/a\/1190000047148676<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"<p>Triton \u662f\u4e00\u79cd\u7528\u4e8e\u5e76\u884c\u7f16\u7a0b\u7684\u8bed\u8a00\u548c\u7f16\u8bd1\u5668\u3002\u5b83\u65e8\u5728\u63d0\u4f9b\u4e00\u4e2a\u57fa\u4e8e Python \u7684\u7f16\u7a0b\u73af\u5883\uff0c\u4ee5\u9ad8\u6548\u7f16\u5199\u81ea\u5b9a\u4e49 [&hellip;]<\/p>\n","protected":false},"author":8,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_jetpack_memberships_contains_paid_content":false,"footnotes":""},"categories":[4],"tags":[],"class_list":["post-10020952","post","type-post","status-publish","format-standard","hentry","category-share"],"jetpack_sharing_enabled":true,"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020952","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/users\/8"}],"replies":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=10020952"}],"version-history":[{"count":0,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020952\/revisions"}],"wp:attachment":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=10020952"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=10020952"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=10020952"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}