triton.language.zeros

triton.language.zeros = JITFunction(triton.language.standard:zeros)

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

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

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