numpy 中的擴展向量（Pytorch 擴展）

Question

我有一個 B x 1 形狀的 numpy 張量，名為Y 。 我正在為 Pytorch 中的以下擴展命令尋找等效的 numpy（ N是整數）：

Y.expand(-1, N)[:, None]

Answer 1

np.tile 呢？

>>> B = 4
>>> N = 3
>>> a = np.arange(0, B).reshape(B, 1)
>>> np.tile(a, (1, N))
array([[0, 0, 0],
       [1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])