dpctl.tensor.ones

dpctl.tensor.ones(shape, dtype=None, order='C', device=None, usm_type='device', sycl_queue=None)[source]

Returns a new dpctl.tensor.usm_ndarray having a specified shape and filled with ones.

Parameters:

  • shape (Tuple[int], int) – Dimensions of the array to be created.

  • 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", or "F") – 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:

Created array initialized with ones.

Return type:

usm_ndarray