python中stdinput是什么_Python torch.std方法代码示例

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

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

示例1: test_SDE

​点赞 6

​

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

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

def test_SDE(self):

def f(x, a, s):

return -a * x

def g(x, a, s):

return s

em = AffineEulerMaruyama((f, g), (0.02, 0.15), Normal(0., 1.), Normal(0., 1.), dt=1e-2, num_steps=10)

model = LinearGaussianObservations(em, scale=1e-3)

x, y = model.sample_path(500)

for filt in [SISR(model, 500, proposal=Bootstrap()), UKF(model)]:

filt = filt.initialize().longfilter(y)

means = filt.result.filter_means

if isinstance(filt, UKF):

means = means[:, 0]

self.assertLess(torch.std(x - means), 5e-2)

开发者ID:tingiskhan，项目名称:pyfilter，代码行数:22，

示例2: forward

​点赞 6

​

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

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

def forward(self, z):

if z.size(-1) == 1:

return z

mu = torch.mean(z, keepdim=True, dim=-1)

sigma = torch.std(z, keepdim=True, dim=-1)

ln_out = (z - mu.expand_as(z)) / (sigma.expand_as(z) + self.eps)

if self.affine:

ln_out = ln_out * self.a_2.expand_as(ln_out) + self.b_2.expand_as(ln_out)

# NOTE(nikita): the t2t code does the following instead, with eps=1e-6

# However, I currently have no reason to believe that this difference in

# implementation matters.

# mu = torch.mean(z, keepdim=True, dim=-1)

# variance = torch.mean((z - mu.expand_as(z))**2, keepdim=True, dim=-1)

# ln_out = (z - mu.expand_as(z)) * torch.rsqrt(variance + self.eps).expand_as(z)

# ln_out = ln_out * self.a_2.expand_as(ln_out) + self.b_2.expand_as(ln_out)

return ln_out

# %%

开发者ID:nikitakit，项目名称:self-attentive-parser，代码行数:23，

示例3: comp

​点赞 6

​

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

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

def comp(self, inpu):

in_mat1 = torch.triu(inpu.repeat(inpu.size(0), 1), diagonal=1)

in_mat2 = torch.triu(inpu.repeat(inpu.size(0), 1).t(), diagonal=1)

comp_first = (in_mat1 - in_mat2)

comp_second = (in_mat2 - in_mat1)

std1 = torch.std(comp_first).item()

std2 = torch.std(comp_second).item()

comp_first = torch.sigmoid(comp_first * (6.8 / std1))

comp_second = torch.sigmoid(comp_second * (6.8 / std2))

comp_first = torch.triu(comp_first, diagonal=1)

comp_second = torch.triu(comp_second, diagonal=1)

return (torch.sum(comp_first, 1) + torch.sum(comp_second, 0) + 1) / inpu.size(0)

开发者ID:technicolor-research，项目名称:sodeep，代码行数:19，

示例4: __call__

​点赞 6

​

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

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

def __call__(self, x):

if not self.auto:

for idx in range(x.shape[0]):

xmean = torch.mean(x[idx, :, :])

xstd = torch.std(x[idx, :, :])

x[idx, :, :] = (x[idx, :, :] - xmean) / xstd

if xstd == 0:

x[idx, :, :] = 0.0

else:

view = x.view(x.shape[0], -1)

length = view.shape[1]

mean = view.mean(dim=1)

var = view.var(dim=1)

self.var = var / (self.count + 1) + self.count / (self.count + 1) * self.var

self.var += self.count / ((self.count + 1) ** 2) * (self.mean - mean) ** 2

self.mean = (self.count * self.mean + view.mean(dim