torch.detach() torch.item()

torch.detach() torch.item()

1. torch.detach()
   用于将当前的tensor从计算图中取出
   通俗点解释就是,经过detach返回的张量将不会进行反向传播计算梯度

2. torch.item()

   用于将单个的tensor值,返回成python中的一个数据形式,多维的tensor就不行了,多维可以用torch.tolist()

   # 单个元素
   x = torch.tensor([1.0])
   x.item()
   >>>1.0
   
   # 多维元素
   a = torch.randn(2, 2)
   a.tolist()
   >>>[[0.012766935862600803, 0.5415473580360413],
    [-0.08909505605697632, 0.7729271650314331]]
   
   a[0,0].tolist()
   >>>0.012766935862600803
   

3. 字典中的.item()
   以列表返回可遍历的(键, 值)元组数组

   dict = {'a':'15岁',
           'b':'14岁',
           'c':'2岁',
           }
   
   # 用法
   for key,values in dict.items():