1分钟理解pytorch的reshape函数中-1表示的意义_万方名的博客 …
https://blog.csdn.net/weixin_42599499/article/details/10589689402.05.2020 · pytorch中reshape函数解析 scar2016的博客 03-242637 1.reshape(m,n) 即: 将矩阵变成 m x n列矩阵 1.1代码: import torch x = torch.arange(12) # 生成一维行向量 ,从1到11x 1.1结果: tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]) 1.2代码: x.reshape(3,4) # 将矩阵x变成 3行4列矩阵 1.2结果: tensor([[ 0, 1, 2, 3], [ 4, 5, 6, 7] 评论1 请先登录后发表评论~ 插入表情 代码片 …