chainer.functions.select_item¶
-
chainer.functions.
select_item
(x, t)[source]¶ Select elements stored in given indices.
This function returns
t.choose(x.T)
, that meansy[i] == x[i, t[i]]
for alli
.Parameters: - x (
Variable
ornumpy.ndarray
orcupy.ndarray
) – Variable storing arrays. A two-dimensional float array. - t (
Variable
ornumpy.ndarray
orcupy.ndarray
) – Variable storing index numbers. A one-dimensional int array. Length of thet
should be equal tox.shape[0]
.
Returns: Variable that holds
t
-th element ofx
.Return type: Example
>>> x = np.array([[0, 1, 2], [3, 4, 5]], 'f') >>> t = np.array([0, 2], 'i') >>> y = F.select_item(x, t) >>> y.shape (2,) >>> y.data array([0., 5.], dtype=float32)
- x (