dpctl.tensor.take¶
- dpctl.tensor.take(x, indices, axis=None, mode='wrap')[source]¶
Takes elements from an array along a given axis at given indices.
- Parameters:
x (usm_ndarray) – The array that elements will be taken from.
indices (usm_ndarray) – One-dimensional array of indices.
axis (int, optional) – The axis along which the values will be selected. If
x
is one-dimensional, this argument is optional. Default:None
.mode (str, optional) –
How out-of-bounds indices will be handled. Possible values are:
"wrap"
: clamps indices to (-n <= i < n
), then wraps negative indices."clip"
: clips indices to (0 <= i < n
).
Default:
"wrap"
.
- Returns:
Array with shape
x.shape[:axis] + indices.shape + x.shape[axis + 1:]
filled with elements fromx
.- Return type: