concatenate
用于拼接多個(gè) ndarray
# `>>` 符號(hào)表示命令行輸出棋电,下同
a = np.arange(12).reshape(3, 2, 2)
b = np.arange(12).reshape(3, 2, 2)
np.concatenate((a, b), axis=0).shape
# >> (6, 2, 2)
np.concatenate((a, b), axis=1).shape
# >> (3, 4, 2)
newaxis
可以用在擴(kuò)充 numpy ndarray 其中某個(gè) axis,如對(duì)于一副 RBG 圖像,擴(kuò)充它的 width
print(np.newaxis)
# >> None
# 假設(shè)一個(gè) numpy array
print(x)
# >> (1, 3, 518, 774)
tmp = x[:, :, -1, :]
print(tmp.shape)
# >> (1, 3, 774)
print(tmp[:, :, np.newaxis, :].shape)
# >> (1, 3, 1, 774)
# 合并
np.concatenate((x, tmp), axis=2).shape
# >> (1, 3, 519, 774)