一、张量
-
张量的属性
- data:张量的数据
- dtype:张量的数据类型(torch.FloatTensor,torch.cuda.FloatTensor)
- shape:张量的形状
- device:张量所在设备(GPU/CPU)
- require_grad:是否需要梯度
- grad:张量的梯度
- grad_fn:张量的function
- is_leaf:是否为叶子节点
-
张量的数据类型
-
共享内存 VS 创建新张量
注意:什么为叶子节点?即用户创建的节点称为叶子节点,如x1,x2,x3,x4等,主要作用是节省内存,因为在梯度反向传播之后,非叶子结点的梯度会被释放。
-
张量的常用创建方法