mars.tensor.full#

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

返回一个给定形状和类型的新张量，填充有 fill_value。

Parameters

  • 形状 (整数 或 整数序列）– 新张量的形状，例如， (2, 3) 或 2。

  • fill_value (标量) – 填充值。

  • dtype (数据类型, 可选) –

    张量的期望数据类型 默认值 None 意味着

    np.array(fill_value).dtype.

  • chunk_size (int 或 tuple 的 int 或 tuple 的 ints, 可选) – 每个维度上所需的块大小

  • gpu (bool, 可选) – 如果为True，则在GPU上分配张量，默认为False

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

Returns

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

Return type

张量

另请参阅

zeros_like

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

ones_like

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

empty_like

返回一个具有输入形状和类型的空张量。

full_like

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

zeros

返回一个新的张量，将值设置为零。

ones

返回一个新的张量，值设置为一。

empty

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

示例

>>> import mars.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]])