pytorch框架语法学习(1)——多维矩阵(tensor)
pytorch框架语法学习(1)——多维矩阵(tensor)
在运用python编程时,总会遇到tensor张量,例如reshape(2,3,4,4)等多维矩阵。下面进行详细讲解。
下面展示一些 内联代码片。
// 生成一个(2,3,4,4)
import torch
mask=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 ·
1351 阅读 ·
1 评论