megengine.functional.ones_like¶

返回一个与输入张量形状相同的全1张量

参数: inp (Union[Tensor, SymbolVar]) – input tensor.
返回类型: Union[Tensor, SymbolVar]
返回: output tensor.

实际案例

import numpy as np
from megengine import tensor
import megengine.functional as F

inp = tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3))
out = F.ones_like(inp)
print(out.numpy())

Outputs:

[[1 1 1]
 [1 1 1]]

megengine.functional.ones

megengine.functional.full