triton.experimental.gluon.language.zeros_like

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

Create a tensor with the same properties as a given tensor, filled with zeros.

Parameters:
  • input (tensor) – Reference tensor to infer default shape, dtype, and layout.

  • shape (Sequence[int], optional) – Target shape. Defaults to input.shape.

  • dtype (dtype, optional) – Target data type. Defaults to input.dtype.

  • layout (DistributedLayout, optional) – Target layout. Defaults to input.layout.

Returns:

A tensor where every element is zero.

Return type:

tensor