triton_language.zeros
triton.language.zeros(shape, dtype
对于给定的shape
和 dtype
,返回 1 个填充了标量值 0 的张量。
参数:
-
shape (tuple of ints) - 新数组的形状,例如 (8,16) 或 (8,)。
-
dtype (DType) - 新数组的数据类型,例如
tl.float16
。
triton.language.zeros(shape, dtype
对于给定的shape
和 dtype
,返回 1 个填充了标量值 0 的张量。
参数:
shape (tuple of ints) - 新数组的形状,例如 (8,16) 或 (8,)。
dtype (DType) - 新数组的数据类型,例如 tl.float16
。