triton.language.zeros_like

triton.language.zeros_like(input)

返回一个与给定张量具有相同形状和类型的零张量。

参数:

input (Tensor) – 输入张量