nvidia.dali.experimental.dynamic.ones_like

nvidia.dali.experimental.dynamic.ones_like(data_like, /, *, batch_size=None, device=None, dtype=None)

Returns new data with the same shape, type and layout as the input array, filled with ones.

Supported backends

• ‘cpu’

Parameters:

data_like (Tensor/Batch) – The input data value to copy the shape, type and layout from.

Keyword Arguments:

• dtype (nvidia.dali.types.DALIDataType, optional) – Overrides the output data type.

• batch_size (int, optional) – The batch size to broadcast input tensors to. Ignored for batch inputs.

• device (device-like, optional) – The device to use for the operation. Must not conflict with the device of the inputs.