maxframe.tensor.full#

maxframe.tensor.full(shape, fill_value, dtype=None, chunk_size=None, gpu=None, order='C')[源代码]#

返回一个给定形状和类型的张量，其元素全部填充为 fill_value。

参数:

  • shape (int or sequence of ints) -- 新张量的形状，例如 (2, 3) 或 2。

  • fill_value (scalar) -- 填充值。

  • dtype (data-type, optional) -- 张量的期望数据类型。默认值 None 表示 np.array(fill_value).dtype。

  • chunk_size (int or tuple of int or tuple of ints, optional) -- 每个维度上的期望块大小

  • gpu (bool, optional) -- 如果为 True，则将张量分配在 GPU 上，默认为 False

  • order ({'C', 'F'}, optional) -- 是否以 C 或 Fortran 连续（行或列优先）顺序在内存中存储多维数据。

返回:

out -- 具有给定形状、数据类型和内存顺序的 fill_value 张量。

返回类型:

Tensor

参见

zeros_like

返回一个形状和类型与输入相同的零张量。

ones_like

返回一个形状和类型与输入相同的全一张量。

empty_like

返回一个形状和类型与输入相同但未初始化的张量。

full_like

用输入的形状和类型填充一个张量。

zeros

返回一个将所有值设为零的新张量。

ones

返回一个将所有值设为一的新张量。

empty

返回一个未初始化的新张量。

示例

>>> import maxframe.tensor as mt

>>> mt.full((2, 2), mt.inf).execute()
array([[ inf,  inf],
       [ inf,  inf]])
>>> mt.full((2, 2), 10).execute()
array([[10, 10],
       [10, 10]])