jax.numpy.r_

jax.numpy.r_ = <jax._src.numpy.index_tricks.RClass object>

沿著第一個軸串聯切片、純量和類陣列物件。

numpy.r_ 的 LAX 後端實作。

參見

jnp.c_：沿著最後一個軸串聯切片、純量和類陣列物件。

範例

以 [start:stop:step] 形式傳遞切片會產生 jnp.arange 物件

>>> jnp.r_[-1:5:1, 0, 0, jnp.array([1,2,3])]
Array([-1,  0,  1,  2,  3,  4,  0,  0,  1,  2,  3], dtype=int32)

step 的虛數值將建立 jnp.linspace 物件，其中包括右端點

>>> jnp.r_[-1:1:6j, 0, jnp.array([1,2,3])]
Array([-1.        , -0.6       , -0.20000002,  0.20000005,
       0.6       ,  1.        ,  0.        ,  1.        ,
       2.        ,  3.        ], dtype=float32)

使用 "axis,dims,trans1d" 形式的字串指令作為第一個引數，以指定串聯軸、最小維度數，以及升級陣列的原始維度在結果陣列的形狀元組中的位置

>>> jnp.r_['0,2', [1,2,3], [4,5,6]] # concatenate along first axis, 2D output
Array([[1, 2, 3],
       [4, 5, 6]], dtype=int32)

>>> jnp.r_['0,2,0', [1,2,3], [4,5,6]] # push last input axis to the front
Array([[1],
       [2],
       [3],
       [4],
       [5],
       [6]], dtype=int32)

trans1d 的負值會將最後一個軸朝形狀元組的開頭偏移

>>> jnp.r_['0,2,-2', [1,2,3], [4,5,6]]
Array([[1],
       [2],
       [3],
       [4],
       [5],
       [6]], dtype=int32)

在扁平輸入上使用特殊指令 "r" 或 "c" 作為第一個引數，以建立具有額外列軸或欄軸的陣列

>>> jnp.r_['r',[1,2,3], [4,5,6]]
Array([[1, 2, 3, 4, 5, 6]], dtype=int32)

>>> jnp.r_['c',[1,2,3], [4,5,6]]
Array([[1],
       [2],
       [3],
       [4],
       [5],
       [6]], dtype=int32)

對於更高維度的輸入 (dim >= 2)，指令 "r" 和 "c" 都會產生相同的結果。