Triton
入门指南
安装
Tutorials
Python API
triton
triton.language
编程模型
创建操作
triton.language.arange
triton.language.cat
triton.language.full
triton.language.zeros
triton.language.zeros_like
zeros_like()
triton.language.cast
形状操作运算
线性代数运算
内存/指针操作
索引操作
数学运算
归约操作
扫描/排序操作
原子操作
随机数生成
迭代器
内联汇编
编译器提示操作
调试操作
triton.testing
Triton 语义
Triton MLIR方言
Triton MLIR方言与操作
编程指南
简介
相关工作
调试Triton
Triton
triton.language
triton.language.zeros_like
查看页面源代码
triton.language.zeros_like
¶
triton.language.
zeros_like
(
input
)
¶
返回一个与给定张量形状和类型相同的全零张量。
Parameters
:
input
(
Tensor
) – 输入张量