[Bug Fix] Fix when tensor with shape [1] and when device in gm.code

[ooooo-create]

PR Category

Bug Fix

Description

1. torch.complie 提取出来的 gm.code 有时会出现 device = device(type='cuda', index=0)，需要先 from torch import device
2. 获取张量的 std info 的时候，存在张量 shape = [1] ，出现 std=nan，直接保存到 weight_meta.py 会导致 nan 未定义，所以改成了 float("nan")
3. 问题模型在 [New Sample] Add 'squim_subjective' Model Computational Graph #83, 这里面是没修改之前的提取

[paddle-bot]

Thanks for your contribution!