pytorch框架语法学习(1)——多维矩阵(tensor)
pytorch框架语法学习(1)——多维矩阵(tensor)在运用python编程时,总会遇到tensor张量,例如reshape(2,3,4,4)等多维矩阵。下面进行详细讲解。下面展示一些 内联代码片。// 生成一个(2,3,4,4)import torchmask=torch.arange(96)mask = mask.reshape(2,3,4,4)print(mask)结果如下:tensor([[[[ 0, 1, 2, 3],[ 4, 5, 6, 7],[ 8,
原创
2021-07-08 23:47:49 ·
1388 阅读 ·
1 评论