triton.language.full

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

Returns a tensor filled with the scalar value for the given shape and dtype.

Parameters:

• shape (tuple of ints) – Shape of the new array, e.g., (8, 16) or (8, )

• value (scalar) – A scalar value to fill the array with

• dtype (tl.dtype) – Data type of the new array, e.g., tl.float16