torch.detach() torch.item()
-
torch.detach()
用于将当前的tensor从计算图中取出
通俗点解释就是,经过detach返回的张量将不会进行反向传播计算梯度 -
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
-
字典中的
.item()
以列表返回可遍历的(键, 值)元组数组dict = {'a':'15岁', 'b':'14岁', 'c':'2岁', } # 用法 for key,values in dict.items():