Numpy,Tensor(Pytorch) 使用对照

矩阵

Numpy	Tensor
a.size	\	矩阵元素数量
a.shape	a.shape	形状大小

属性

操作

Numpy	Tensor
np.expand_dims(a, axis=1)	a.unsqueeze(1)	升维
np.squeeze(a)	a.squeeze()	降维
np.concatenate((a,b),axis=1)	torch.cat((a,b),dim=1)	拼接
np. tile(a,2)	torch.repeat(2)	扩展
a.repeat(3,2)	\	分别复制