pytorch tensor 张量中数值替换，极小数值极大数值，nan替换，并让他支持梯度

zhangfeng1133

于 2024-09-09 16:02:23 发布

阅读量266

点赞数 1

文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/zhangfeng1133/article/details/142061274

版权

/root/.local/lib/python3.10/site-packages/pyro/infer/autoguide/guides.py

import random

        loc = self.loc
	scale = self.scale
	# 避免原地操作，而是创建一个新的变量, 支持梯度
        loc = torch.where(loc < 0.0001, 0.00001, loc)
        #loc[loc > 10000] = random.randint(1, 1000)
        #print(scale)
        #nan_mask = torch.isnan(loc)
        #loc[nan_mask] = 0.00001
        loc = torch.nan_to_num(loc, nan=0.00001)

	# 避免原地操作，而是创建一个新的变量, 支持梯度
        scale = torch.where(scale < 0.0001, 0.00001, scale)
        #scale[scale > 10000] = random.randint(1, 1000)
        #print(scale)
        #nan_mask = torch.isnan(scale)
        #scale[nan_mask] = 0.00001
        scale = torch.nan_to_num(scale, nan=0.00001)