triton.language.full

triton.language.full(shape, value, dtype)

返回一个根据给定的shape和dtype填充标量值的张量。

Parameters:

• shape (tuple of ints) – 新数组的形状，例如 (8, 16) 或 (8, )

• value (标量) – 用于填充数组的标量值

• dtype (tl.dtype) – 新数组的数据类型，例如 tl.float16