megengine.functional.ones

ones(shape, dtype='float32', device=None)

返回一个具有给定形状的全1张量。

参数

inp – 输入张量。

返回类型

Tensor

返回

输出的全0张量。

例如：

import megengine.functional as F

out = F.ones((2, 1))
print(out.numpy())

输出：

[[1.]
 [1.]]

megengine.functional.zeros_like megengine.functional.ones_like