| matlab | python |
|---|---|
| randn(1，5) | np.random.randn(5) |
python中X[:, [0]]与X[:, 0]不同
X = np.array([[1,2,3], [4,5,6], [7,8,9]])
A = X[:, [0]]
B = X[:, 0]
python中拼接函数常用的有以下几种
- Numpy中stack()，hstack()，vstack()
- pytorch中stack()，cat()
matlab中cat()函数与python中torch.cat()函数并不相同，torch.cat是将两个张量（tensor）拼接在一起，而要实现矩阵的拼接应该是在python中Numpy中stack()，hstack()，vstack()，concatenate（）函数
| matlab | python |
|---|---|
| cat(1,A,B) | np.vstack(A,B) 或np.concatenate((A,B)) |
| cat(2,A,B) | np.hstack(A,B)或np.concatenate((A,B),axis=1) |
| cat(3,A,B) | np.stack([A,B],axis=2) |
