Python mxnet.contrib.ndarray.index_array用法及代碼示例

用法:
mxnet.contrib.ndarray.index_array(data=None, axes=_Null, out=None, name=None, **kwargs)

參數：

data：(NDArray) - 輸入數據
axes：(Shape or None, optional, default=None) - 要包含在索引數組中的軸。支持負值。
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

返回輸入數組的索引數組。

對於形狀為 \((d_1, d_2, ..., d_n)\) 的輸入數組，index_array 返回一個 \((d_1, d_2, ..., d_n, n)\) 數組 idx ，其中 \(idx[i_1, i_2, ..., i_n, :] = [i_1, i_2, ..., i_n]\) 。

此外，當指定參數 axes 時，idx 將是一個 \((d_1, d_2, ..., d_n, m)\) 數組，其中 m 是 axes 的長度，並且以下等式將成立：\(idx[i_1, i_2, ..., i_n, j] = i_{axes[j]}\) 。

例子：

x = mx.nd.ones((3, 2))

mx.nd.contrib.index_array(x) = [[[0 0]
                                 [0 1]]

                                [[1 0]
                                 [1 1]]

                                [[2 0]
                                 [2 1]]]

x = mx.nd.ones((3, 2, 2))

mx.nd.contrib.index_array(x, axes=(1, 0)) = [[[[0 0]
                                               [0 0]]

                                              [[1 0]
                                               [1 0]]]


                                             [[[0 1]
                                               [0 1]]

                                              [[1 1]
                                               [1 1]]]


                                             [[[0 2]
                                               [0 2]]

                                              [[1 2]
                                               [1 2]]]]

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.contrib.ndarray.index_array。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：