pytorch常用函数之torch.randn()解读

更新时间：2023年02月02日 09:23:49 作者：土豆豆豆豆豆

这篇文章主要介绍了pytorch常用函数之torch.randn()解读，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

pytorch常用函数torch.randn()

torch.randn(*sizes, out=None) → Tensor

功能：从标准正态分布（均值为0，方差为1）中抽取的一组随机数。返回一个张量

sizes (int…) - 整数序列，定义输出张量的形状
out (Tensor, optinal) - 结果张量

eg：

random = torch.randn(2, 3）
out:  0.5419 0.1594 -0.0413
        -2.7937 0.9534 0.4561

pytorch torch.chunk(tensor, chunks, dim)

说明：在给定的维度上讲张量进行分块。

参数：

tensor(Tensor) -- 待分块的输入张量
chunks(int) -- 分块的个数
dim(int) -- 维度，沿着此维度进行分块

>>> x = torch.randn(3, 3)
>>> x
tensor([[ 1.0103,  2.3358, -1.9236],
        [-0.3890,  0.6594,  0.6664],
        [ 0.5240, -1.4193,  0.1681]])
>>> torch.chunk(x, 3, dim=0)
(tensor([[ 1.0103,  2.3358, -1.9236]]), tensor([[-0.3890,  0.6594,  0.6664]]), tensor([[ 0.5240, -1.4193,  0.1681]]))
>>> torch.chunk(x, 3, dim=1)
(tensor([[ 1.0103],
        [-0.3890],
        [ 0.5240]]), tensor([[ 2.3358],
        [ 0.6594],
        [-1.4193]]), tensor([[-1.9236],
        [ 0.6664],
        [ 0.1681]]))
>>> torch.chunk(x, 2, dim=1)
(tensor([[ 1.0103,  2.3358],
        [-0.3890,  0.6594],
        [ 0.5240, -1.4193]]), tensor([[-1.9236],
        [ 0.6664],
        [ 0.1681]]))

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

关于pymysql模块的使用以及代码详解
在本篇文章里小编给大家整理的是关于关于pymysql模块的使用以及代码详解，有兴趣的朋友们学习下。
2019-09-09
python在windows命令行下输出彩色文字的方法
这篇文章主要介绍了python在windows命令行下输出彩色文字的方法,涉及Python文字特效操作技巧,需要的朋友可以参考下
2015-03-03
python基于Selenium的web自动化框架
这篇文章主要介绍了python基于Selenium的web自动化框架，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-07-07
一篇文章带你了解Python和Java的正则表达式对比
正则表达式有元字符及不同组合来构成，通过巧妙的构造正则表达式可以匹配任意字符串，并完成复杂的字符串处理任务，希望本片文章能给你带来帮助
2021-09-09
在Python中使用AOP实现Redis缓存示例
本篇文章主要介绍了在Python中使用AOP实现Redis缓存示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-07-07
Python django使用多进程连接mysql错误的解决方法
这篇文章主要介绍了Python django使用多进程连接mysql错误的解决方法，详细的介绍了解决方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-10-10
python实现朴素贝叶斯分类器
这篇文章主要为大家详细介绍了python实现朴素贝叶斯分类器，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-03-03
python使用redis模块来跟redis实现交互
这篇文章主要介绍了python使用redis模块来跟redis实现交互，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下
2022-06-06
Python实现曲线的肘部点检测详解
肘部法则是经常使用的法则。很多时候，可以凭人工经验去找最优拐点，但有时需要自动寻找拐点。本文为大家介绍了Python实现曲线的肘部点检测的方法，希望对大家有所帮助
2023-02-02
Python帮你识破双11的套路
一年一度的“双十一”又要来了，很多人已经开始摩拳擦掌，毕竟几天之后手还在不在就不好说了。看看Python帮你识破双11的套路,需要的朋友可以参考下
2019-11-11