torch计算

max和min计算注意：

　　torch.max和torch.min返回的是两个最大值和对应下标

　　当只要最大值的数据时候，需要在变量后加上【0】

　　比如：

　　S_max = torch.max(image_gamma,dim = 0,keepdim = False)[0]

torch中的逻辑运算：

　　a=torch.randn(2,2,2)

　　b=torch.randn(2,2,2)

　　c=(a==b)

　　返回的结果c中是一个2*2*2的bool张量，每个位置分别对应a和b对应元素是否相等，相等为1不等为0

　　d[c]=1.0可以对c中相应位置1的进行修改

　　d[~c]=1.0可以对c中相应位置为0的地方进行修改

　　逻辑与运算：

　　当a、d和b是都是2*2*2的bool张量时

　　c = （（a>d） & (b>d)）

　　与符号为&，不能是and

　　a>d和b>d都需要用括号括起来，否则会报错

torch计算

原文：https://www.cnblogs.com/baitian963/p/13329332.html