megengine.functional.ones_like

ones_like(inp)[源代码]

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

返回类型

Union[Tensor, SymbolVar]

参数

inp (Union[megengine.tensor.Tensor, megengine.core._imperative_rt.core2.SymbolVar]) –