曲靖市网站建设_网站建设公司_Django_seo优化
2026/1/20 12:01:50 网站建设 项目流程

Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境,以高效编写自定义 DNN 计算内核,并能够在现代 GPU 硬件上以最大吞吐量运行。

更多 Triton 中文文档可访问 →triton.hyper.ai/

triton.language.where(condition, x, y)

根据condition返回来自xy元素的张量。
注意:无论condition的值是什么,xy总是会被求值。
如果希望避免意外的内存操作,请使用triton.loadtriton.store中的mask参数。
xy的形状都会被广播到condition的形状。xy必须具有相同的数据类型。
参数**:**

  • conditiontriton.bool 的块)- 当为 True(非零)时,产生 x,否则产生 y。
  • x- 在条件为 True 的索引处选择的值。
  • y- 在条件为 False 的索引处选择的值。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询