megengine.functional.ones_like¶
- ones_like(inp)[source]¶
Returns a tensor filled with ones with the same shape and data type as input tensor.
- Parameters
inp (Tensor) – input tensor from which to derive the output tensor shape.
- Return type
- Returns
a tensor having the same shape as input tensor and filled with ones.
Examples
>>> x = F.arange(6, dtype='int32').reshape(2, 3) >>> F.ones_like(x) Tensor([[1 1 1] [1 1 1]], dtype=int32, device=xpux:0)