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
>>> out = F.full([2,3], 1.5)
>>> out.numpy()
array([[1.5, 1.5, 1.5],
       [1.5, 1.5, 1.5]], dtype=float32)