{"id":10020968,"date":"2025-08-21T21:25:38","date_gmt":"2025-08-21T13:25:38","guid":{"rendered":"https:\/\/itmax.cn\/?p=10020968"},"modified":"2025-08-21T21:25:38","modified_gmt":"2025-08-21T13:25:38","slug":"%e3%80%90triton-%e6%95%99%e7%a8%8b%e3%80%91triton_language-expand_dims","status":"publish","type":"post","link":"https:\/\/itmax.cn\/?p=10020968","title":{"rendered":"\u3010Triton \u6559\u7a0b\u3011triton_language.expand_dims"},"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=rrhj2UwpiNAm1N%2Fu2wE1%2BA%3D%3D.oAh0wlTzCs7PYHbrBuZZcjBpjP4edNhUV3g5UFPZAPM%3D\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">https:\/\/triton.hyper.ai\/<\/a><\/p>\n<pre><code>triton.language.expand_dims(input, axis)<\/code><\/pre>\n<p>\u901a\u8fc7\u63d2\u5165\u65b0\u7684\u957f\u5ea6\u4e3a 1 \u7684\u7ef4\u5ea6\u6765\u6269\u5c55\u5f20\u91cf\u7684\u5f62\u72b6\u3002<\/p>\n<p>\u8f74\u7d22\u5f15\u662f\u76f8\u5bf9\u4e8e\u751f\u6210\u7684\u5f20\u91cf\u800c\u8a00\u7684\uff0c\u56e0\u6b64\u5bf9\u4e8e\u6bcf\u4e2a\u8f74\uff0c<code>result.shape[axis]<\/code>\u00a0\u5c06\u4e3a 1\u3002<\/p>\n<p><strong>\u53c2\u6570<\/strong>\uff1a<\/p>\n<ul>\n<li><strong>input<\/strong>\u00a0(<em>tl.tensor<\/em>) &#8211; \u8f93\u5165\u5f20\u91cf\u3002<\/li>\n<li><strong>axis<\/strong>\u00a0(<em>int<\/em>\u00a0|\u00a0<em>Sequence[int]<\/em> ) &#8211; \u8981\u6dfb\u52a0\u65b0\u8f74\u7684\u7d22\u5f15\u3002<\/li>\n<\/ul>\n<p>\u8be5\u51fd\u6570\u4e5f\u53ef\u4f5c\u4e3a\u00a0<code>tensor<\/code>\u00a0\u7684\u6210\u5458\u51fd\u6570\u8c03\u7528\uff0c\u4f7f\u7528\u00a0<code>x.expand_dims(...)<\/code>\u00a0\u800c\u4e0d\u662f\u00a0<code>expand_dims(x, ...)<\/code>\u3002<\/p>\n<p><a href=\"https:\/\/segmentfault.com\/a\/1190000047175266\">https:\/\/segmentfault.com\/a\/1190000047175266<\/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":1,"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-10020968","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\/10020968","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\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=10020968"}],"version-history":[{"count":0,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020968\/revisions"}],"wp:attachment":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=10020968"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=10020968"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=10020968"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}