torch.bernoulli 的使用方法（附代码示例）

功能解释
参数
代码示例

功能解释

*torch.bernoulli(input, , generator=None, out=None) → Tensor
从伯努利分布中提取二进制随机数（0或1），输入张量应为包含用于绘制二进制随机数的概率的张量。因此，输入中的所有值都必须在以下范围内(0,1)。
第i个元素的输出tensor将要根据输入的第i个概率值，来决定是否生成1值。
输出tensor仅仅只有0或1值，并且其形状和输入的Input的形状是一样的。
输出out可以是一个整型类型，但是输入必须是浮点型数据类型。

参数

 input (Tensor)-对于伯努利分布的输入概率值。generator (torch.Generator, optional) –为了采样，一个伪随机数生成器。out(Tensor, optional) –输出的tensor。

代码示例

这里是代码实现部分

>>> a = torch.empty(3, 3).uniform_(0, 1)  # generate a uniform random matrix with range [0, 1]
>>> a
tensor([[ 0.1737,  0.0950,  0.3609],[ 0.7148,  0.0289,  0.2676],[ 0.9456,  0.8937,  0.7202]])
>>> torch.bernoulli(a)
tensor([[ 1.,  0.,  0.],[ 0.,  0.,  0.],[ 1.,  1.,  1.]])>>> a = torch.ones(3, 3) # probability of drawing "1" is 1
>>> torch.bernoulli(a)
tensor([[ 1.,  1.,  1.],[ 1.,  1.,  1.],[ 1.,  1.,  1.]])
>>> a = torch.zeros(3, 3) # probability of drawing "1" is 0
>>> torch.bernoulli(a)
tensor([[ 0.,  0.,  0.],[ 0.,  0.,  0.],[ 0.,  0.,  0.]])

torch.bernoulli 的使用方法（附代码示例）相关推荐

基于图数据库、图算法、图神经网络的 ID Resolution/ID Mapping 大数据分析方法与代码示例
本文首发于 NebulaGraph 公众号 "本文是一个基于 NebulaGraph 上的图数据库.图算法.图神经网络的 ID Resolution/ID Mapping 方法综述,除了基本 ...
Spring源码深度解析，Spring源码以及Bean的生命周期（五）（附代码示例：）
五)Bean 的生命周期,创建---初始化---销毁的过程目录五)Bean 的生命周期,创建---初始化---销毁的过程一 , 指定初始化方法 init-method 方法二 ,指定销毁 ...
java重写的代码_java tostring方法重写代码示例
当需要将一个对象输出到显示器时,通常要调用他的toString()方法,将对象的内容转换为字符串.java中的所有类默认都有一个toString()方法默认情况下 System.out.printl ...
java实现生日提醒_asp实现的可以提醒生日的几种方法附代码
asp实现的可以提醒生日的几种方法附代码更新时间:2008年06月20日 08:44:52 作者: asp提醒生日的原理一般就是把用户的出生日期和当前日期对比,如果少于几天则提醒显示,下面是几种 ...
html视频海报代码,如何在网站头部添加视频海报？添加视频海报的方法（代码示例）...
本篇文章给大家带来的内容是介绍如何在网站头部添加视频海报?添加视频海报的方法(代码示例) .有一定的参考价值,有需要的朋友可以参考一下,希望对你们有所帮助. 给网站头部添加视频海报的代码: 视频海报 ...
python怎么清除代码_Python如何清空列表？清空列表的4种方法（代码示例）
通过Python语言提供的不同构造的方法可以有许多方法来清空列表,下面本篇文章就来给带大家介绍Python中4种清空列表的方法,希望对大家有所帮助. 方法1:重新初始化列表可以重初始化列表,初始化该 ...
php 引入echarts路径,angular4中引入echarts的方法（代码示例）
本篇文章给大家带来的内容是关于angular4中引入echarts的方法(代码示例),有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助. 1.安装ngx-echarts npm insta ...
设计模式——七大原则（附代码示例）
一. 设计模式概念对接口编程而不是对实现编程:优先使用对象组合而不是继承二. 设计模式总览 1. 创建型模式(Creational Patterns):(5) 单例(Singlet ...
Java强、软、弱、虚四大引用（附代码示例）
一. 整体架构 Reference: 强引用 SoftReference: 软引用 WeakReference: 弱引用 PhantomReference: 虚引用二. 强引用概念当内存不足,J ...

torch.bernoulli 的使用方法（附代码示例）

torch.bernoulli 的使用方法（附代码示例）

功能解释

参数

代码示例

torch.bernoulli 的使用方法（附代码示例）相关推荐

最新文章

热门文章