megengine.functional.full

full(shape, value, dtype=None, device=None)

返回一个新创建的拥有指定shape且被value填充的tensor

参数

• shape (Union[int, tuple, list]) – 输出张量的形状。

• value (Union[bool, int, float, Tensor]) – 填充值

• dtype – 输出tensor的数据类型，当dtype为None时，输出类型同value的数据类型，默认为None

• device – 存放新创建tensor的设备， 默认为None

返回类型

Tensor

返回

一个全value的tensor

实际案例

import numpy as np
import megengine.functional as F

out = F.full([2,3], 1.5)
print(out.numpy())

输出：

[[1.5 1.5 1.5]
 [1.5 1.5 1.5]]

上一页

megengine.functional.ones_like

下一页

megengine.functional.full_like