zeros_like¶
- nvtripy.zeros_like(input: Tensor, dtype: dtype | None = None) Tensor [source]¶
Creates a Tensor with all elements set to 0 of the same shape as the input tensor.
- Parameters:
- Returns:
[dtype=T2] A tensor of the same shape as the input with all elements set to 0.
- Return type:
- DATA TYPE CONSTRAINTS:
Example
1input = tp.iota([2, 3], dtype=tp.float32) 2output = tp.zeros_like(input)
Local Variables¶>>> input tensor( [[0, 0, 0], [1, 1, 1]], dtype=float32, loc=gpu:0, shape=(2, 3)) >>> output tensor( [[0, 0, 0], [0, 0, 0]], dtype=float32, loc=gpu:0, shape=(2, 3))
See also