triton.experimental.gluon.language.zeros

triton.experimental.gluon.language.zeros(shape, dtype, layout=None)

Create a tensor filled with zeros.

Parameters:
  • shape (Sequence[int]) – The shape of the tensor.

  • dtype (dtype) – The data type for the tensor.

  • layout (Optional[DistributedLayout]) – The distributed layout of the tensor, defaults to AutoLayout().

Returns:

A tensor where every element is zero.

Return type:

tensor