网站首页编程语言正文

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

作者：土豆豆豆豆豆更新时间： 2023-04-01 编程语言

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]]))

总结

原文链接：https://blog.csdn.net/JannyYang/article/details/100514037

上一篇：路由react-router-dom的基本使用教程_Reac
下一篇：JQuery动态生成的按钮无法触发问题及完美解决方法_jqu

网站首页编程语言正文

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

目录

pytorch常用函数torch.randn()

pytorch torch.chunk(tensor, chunks, dim)

总结

相关推荐

网站首页 编程语言 正文

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

目录

pytorch常用函数torch.randn()

pytorch torch.chunk(tensor, chunks, dim)

总结

相关推荐

网站首页编程语言正文