triton.language.full

triton.language.full(shape, value, dtype, _semantic=None)

返回一个填充了指定标量值、具有给定 shape（形状）和 dtype（数据类型）的张量。

参数:

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

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

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