张量
1. 什么是张量(Tensors)
张量与Numpy的ndarray相同都表示的是一个多维的矩阵。 与ndarray的最大区别就是,PyTorch的Tensor可以在 GPU 上运行,而 numpy 的 ndarray 只能在 CPU 上运行,在GPU上运行大大加快了运算速度。
2. 成员方法
3. 难点解析
- x.view():类似于numpy中的
reshape
,但更高级一点,如果只知道列,那么第一个参数写-1
比如:如果x为已知列为10,那么就可以这么分割:1
2
3
4
5tensor([[0.1483, 0.6744, 0.9375],
[0.4576, 0.2121, 0.5305],
[0.8123, 0.9203, 0.2041],
[0.4643, 0.2020, 0.3629],
[0.3258, 0.1127, 0.2111]])z = x.view(-1, 10)