Tensor和Variable(笔记)

灿烂世界 9 0 PDF 2021-01-15 22:01:25

1.(以前)Variable是torch.autograd中的数据类型,主要用于封装Tensor,进行自动求导: data:被包装的tensor grad:data的梯度 grad_fn:创建Tensor的Function,是自动求导的关键 requires_grad:指示是否需要梯度 is_leaf:指示是否是叶子节点(张量) 2.(现在)Variable并入Tensor dtype:张量的数据类型,如torch.FloatTensor,torch.cuda.FoatTensor shape

用户评论
请输入评论内容
评分:
暂无评论