megengine.functional.full¶

full(shape, value, *, dtype=None, device=None)[源代码]¶

返回一个新的拥有指定 shape 且全部元素均为指定值的张量。

参数

关键字参数

dtype (Tensor.dtype, optional) – 输出张量的数据类型。如果 dtype 为 None, 则输出张量的数据类型必须从 value 推导出来。如果 value 是一个 int, 则输出张量的默认数据类型为整型。如果 value 是一个 float, 则输出张量的默认数据类型为浮点类型。如果 value 是一个 bool, 则输出张量的默认数据类型为布尔类型。
device (Tensor.device, optional) – 被创建的向量所处的设备

返回类型

Tensor

返回

一个全value的tensor

实际案例

>>> F.full((2, 3), 6)
Tensor([[6 6 6]
 [6 6 6]], dtype=int32, device=xpux:0)

megengine.functional.ones_like

megengine.functional.full_like