linux随笔之reshape()

1、说明:

  reshape()和numpy.reshape()函数的作用是,重塑的数组的shape。

2、注意:(参考链接1:Python中reshape函数参数-1的意思?

  python默认是按行取元素。

  参数-1,表示模糊reshape的意思。

  比如:reshape(-1,3),固定3列 多少行不知道。

3、实验代码:

  

1 import numpy as np
2 import torch
3 a = [[1,2,3],[4,5,6]] #定义一个数组
4 b = torch.tensor(a)    #初始化一个tensor
5 print("原始数组为:\n{0}\n\n".format(b))
6 print("reshape(-1),(就是把任何shape的tensor拉平):\n{0}\n\n".format(b.reshape(-1)))#string.format(var)是python里的格式化输出函数
7 print("reshape(-1,2),(reshape成三行两列):\n{0}\n\n".format(b.reshape(-1,2)))

 

4、实验结果(jupyter notebook)

原始数组为:
tensor([[1, 2, 3],
        [4, 5, 6]])


reshape(-1),(就是把任何shape的tensor拉平):
tensor([1, 2, 3, 4, 5, 6])


reshape(-1,2),(reshape成三行两列):
tensor([[1, 2],
        [3, 4],
        [5, 6]])

 

linux随笔之reshape()

上一篇:shell脚本入门基础


下一篇:Mac系统上查看端口占用和释放端口教程