mars.tensor.ones#

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

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

Parameters

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

  • dtype (数据类型, 可选) – 期望的张量数据类型，例如 mt.int8。 默认是 mt.float64。

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

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

  • order ({'C', 'F'}, 可选, 默认: C) – 是否将多维数据以行优先 (类似C的风格)或列优先(类似Fortran的风格)的顺序存储在 内存中。

Returns

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

Return type

张量

另请参阅

zeros, ones_like

示例

>>> import mars.tensor as mt

>>> mt.ones(5).execute()
array([ 1.,  1.,  1.,  1.,  1.])

>>> mt.ones((5,), dtype=int).execute()
array([1, 1, 1, 1, 1])

>>> mt.ones((2, 1)).execute()
array([[ 1.],
       [ 1.]])

>>> s = (2,2)
>>> mt.ones(s).execute()
array([[ 1.,  1.],
       [ 1.,  1.]])