JERRYLSU

Numpy [None, …]

None在对应位置增加一个维度(类似于unsqueeze(axis=))

...等价于[:,:,:]

import numpy as np

arr = np.random.randn(5, 10)
arr.shape

(5, 10)

arr[None, ...].shape

(1, 5, 10)

arr[..., None].shape

(5, 10, 1)

arr[:, None, :].shape

(5 …

Numpy np.where

Reason is the light and the light of life.

Numpy Indexing

import numpy as np

索引数组

# 索引数组必须是整数类型。索引数组种 …