mxnet.np.ones_like

ones_like(a, dtype=None, order='C', device=None, out=None)

Return an array of ones with the same shape and type as a given array.

Parameters

  • a (ndarray) – The shape and data-type of a define these same attributes of the returned array.

  • dtype (data-type, optional) – Overrides the data type of the result. Temporarily do not support boolean type.

  • order ({'C'}, optional) – Whether to store multidimensional data in C- or Fortran-contiguous (row- or column-wise) order in memory. Currently only supports C order.

  • device (Device, optional) – Device context on which the memory is allocated. Default is mxnet.device.current_device().

  • out (ndarray or None, optional) – A location into which the result is stored. If provided, it must have the same shape and dtype as input ndarray. If not provided or None, a freshly-allocated array is returned.

Returns

out – Array of ones with the same shape and type as a.

Return type

ndarray

See also

empty_like()

Return an empty array with shape and type of input.

zeros_like()

Return an array of zeros with shape and type of input.

full_like()

Return a new array with shape of input filled with value.

ones()

Return a new array setting values to one.

Examples

>>> x = np.arange(6)
>>> x = x.reshape((2, 3))
>>> x
array([[0., 1., 2.],
       [3., 4., 5.]])
>>> np.ones_like(x)
array([[1., 1., 1.],
       [1., 1., 1.]])
>>> np.ones_like(x, int)
array([[1, 1, 1],
       [1, 1, 1]], dtype=int64)
>>> y = np.arange(3, dtype=float)
>>> y
array([0., 1., 2.], dtype=float64)
>>> np.ones_like(y)
array([1., 1., 1.], dtype=float64)