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)同理