python torch exp_Python torch.add方法代码示例

本文整理汇总了Python中torch.add方法的典型用法代码示例。如果您正苦于以下问题：Python torch.add方法的具体用法？Python torch.add怎么用？Python torch.add使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块torch的用法示例。

在下文中一共展示了torch.add方法的16个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: _transform

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import add [as 别名]

def _transform(x, mat, maxmin):

rot = mat[:,0:3]

trans = mat[:,3:6]

x = x.contiguous().view(-1, x.size()[1] , x.size()[2] * x.size()[3])

max_val, min_val = maxmin[:,0], maxmin[:,1]

max_val, min_val = max_val.contiguous().view(-1,1), min_val.contiguous().view(-1,1)

max_val, min_val = max_val.repeat(1,3), min_val.repeat(1,3)

trans, rot = _trans_rot(trans, rot)

x1 = torch.matmul(rot,x)

min_val1 = torch.cat((min_val, Variable(min_val.data.new(min_val.size()[0], 1).fill_(1))), dim=-1)

min_val1 = min_val1.unsqueeze(-1)

min_val1 = torch.matmul(trans, min_val1)

min_val = torch.div( torch.add(torch.matmul(rot, min_val1).squeeze(-1), - min_val), torch.add(max_val, - min_val))

min_val = min_val.mul_(255)

x = torch.add(x1, min_val.unsqueeze(-1))

x = x.contiguous().view(-1,3, 224,224)

return x

开发者ID:microsoft，项目名称:View-Adaptive-Neural-Networks-for-Skeleton-based-Human-Action-Recognition，代码行数:26，

示例2: get_loss

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import add [as 别名]

def get_loss(pred, y, criterion, mtr, a=0.5):

"""

To calculate loss

:param pred: predicted value

:param y: actual value

:param criterion: nn.CrossEntropyLoss

:param mtr: beta matrix

"""

mtr_t = torch.transpose(mtr, 1, 2)

aa = torch.bmm(mtr, mtr_t)

loss_fn = 0

for i in range(aa.size()[0]):

aai = torch.add(aa[i, ], Variable(torch.neg(torch.eye(mtr.size()[1]))))

loss_fn += torch.trace(torch.mul(aai, aai).data)

loss_fn /= aa.size()[0]

loss = torch.add(criterion(pred, y), Variable(torch.FloatTensor([loss_fn * a])))

return loss

开发者ID:BarnesLab，项目名称:Patient2Vec，代码行数:19，

示例3: forward

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import add [as 别名]

def forward(self, x):

if not self.equalInOut:

x = self.relu1(self.bn1(x))

else:

out = self.relu1(self.bn1(x))

if self.equalInOut:

out = self.relu2(self.bn2(self.conv1(out)))

else:

out = self.relu2(self.bn2(self.conv1(x)))

if self.droprate > 0:

out = F.dropout(out, p=self.droprate, training=self.training)

out = self.conv2(out)

if