megengine.functional.ones

ones(shape, dtype='float32', device=None)[源代码]

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

参数

inp – 输入张量。

返回

输出的全0张量。

例如:

import megengine.functional as F

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

输出:

[[1.]
 [1.]]