Pytorch torch.rsqrt()的简单用法

官方文档：https://pytorch.org/docs/master/generated/torch.rsqrt.html

简单来说就是对每个元素取平方根后再取倒数，即：
out i=1input i\text { out }_{i}=\frac{1}{\sqrt{\text { input }_{i}}} out i= input i1
是元素级的操作，并不会影响如维度等因素。

例如：

import torchx = torch.Tensor([[1, 4],[16, 25]])
y = torch.rsqrt(x)
print(y)

输出：

tensor([[1.0000, 0.5000],[0.2500, 0.2000]])

Pytorch torch.rsqrt()的简单用法相关推荐

Pytorch torch.topk()的简单用法
官方文档:https://pytorch.org/docs/stable/generated/torch.topk.html?highlight=topk#torch.topk 由于numpy本身是没 ...
Pytorch torch.mean()的简单用法
简单来说就是求平均数. 比如以下的三种简单情况: import torchx1 = torch.Tensor([1, 2, 3, 4]) x2 = torch.Tensor([[1],[2],[3], ...
Pytorch torch.manual_seed()的简单用法
简单来说就是指定随机数种子(种子可以是个任意int值),用来生成伪随机数: import torchtorch.manual_seed(1) print(torch.rand(1)) print(to ...
Pytorch torch.nonzero()的简单用法
官方文档:https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/ 顾名思义,返回非零元素的索引: 先从一维的看起: ...
Pytorch nn.DataParallel()的简单用法
简单来说就是使用单机多卡进行训练. 一般来说我们看到的代码是这样的: net = XXXNet() net = nn.DataParallel(net) 这样就可以让模型在全部GPU上训练. 方法定义 ...
Pytorch transforms.Resize()的简单用法
简单来说就是调整PILImage对象的尺寸,注意不能是用io.imread或者cv2.imread读取的图片,这两种方法得到的是ndarray. 将图片短边缩放至x,长宽比保持不变: transfor ...
pytorch tensor查找0_Pytorch简单教程
2019 年,ML 框架之争只剩两个实力玩家:PyTorch 和 TensorFlow.研究者大批涌向 PyTorch,而业界的首选仍然是 TensorFlow. Pytorch和TensorFlow ...
如何使用PyTorch torch.max（）
In this article, we'll take a look at using the PyTorch torch.max() function. 在本文中,我们将介绍如何使用PyTorch ...
Pytorch中的collate_fn函数用法
Pytorch中的collate_fn函数用法官方的解释: Puts each data field into a tensor with outer dimension batch size ...

Pytorch torch.rsqrt()的简单用法

Pytorch torch.rsqrt()的简单用法相关推荐

最新文章

热门文章