dpctl.tensor.ones_like

dpctl.tensor.ones_like(x, /, *, dtype=None, order='K', device=None, usm_type=None, sycl_queue=None)[source]

Returns a new dpctl.tensor.usm_ndarray filled with ones and having the same shape as the input array x.

Parameters:
  • x (usm_ndarray) – Input array from which to derive the output array shape

  • dtype (optional) – data type of the array. Can be typestring, a numpy.dtype object, numpy char string, or a NumPy scalar type. Default: None

  • order ("C", "F", "A", or "K") – memory layout for the array. Default: "C"

  • device (optional) – array API concept of device where the output array is created. device can be None, a oneAPI filter selector string, an instance of dpctl.SyclDevice corresponding to a non-partitioned SYCL device, an instance of dpctl.SyclQueue, or a dpctl.tensor.Device object returned by dpctl.tensor.usm_ndarray.device. Default: None

  • usm_type ("device", "shared", "host", optional) – The type of SYCL USM allocation for the output array. Default: "device"

  • sycl_queue (dpctl.SyclQueue, optional) – The SYCL queue to use for output array allocation and copying. sycl_queue and device are complementary arguments, i.e. use one or another. If both are specified, a TypeError is raised unless both imply the same underlying SYCL queue to be used. If both are None, a cached queue targeting default-selected device is used for allocation and population. Default: None

Returns:

New array initialized with ones.

Return type:

usm_ndarray