Numpy的reshape函数 reshape(-1,x) reshape(x,-1)含义

1. 将tensor类型转变为numpy类型:

a = torch.ones(5)
b = a.numpy()

2.reshape函数的意思是将numpy类型的数据 重新 改变格式:

一般用法:numpy.arange(n).reshape(a, b); 依次生成n个自然数,并且以a行b列的数组形式显示:

In [1]: 
np.arange(16).reshape(2,8) #生成16个自然数,以2行8列的形式显示

Out[1]: 
array([[ 0,  1,  2,  3,  4,  5,  6,  7],
       [ 8,  9, 10, 11, 12, 13, 14, 15]])

 参数包含-1 时:

mat (or array).reshape(c, -1);  必须是矩阵格式或者数组格式才能使用 .reshape(c, -1) 函数

表示将此矩阵或者数组重组,以 c行d列的形式表示,行数固定为c ,列数需要numpy自己计算

.reshape(-1, d)同理
 

你可能感兴趣的:(Python学习,numpy)