chainer.functions.select_item

chainer.functions.select_item(x, t)

Select elements stored in given indices.

This function returns t.choose(x.T), that means y[i] == x[i, t[i]] for all i.

Parameters:

• x (Variable) – Variable storing arrays.
• t (Variable) – Variable storing index numbers.

Returns:

Variable that holds t-th element of x.

Return type:

Variable