当前位置:主页 > python教程 > pytorch函数之torch.randn()

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

发布:2023-04-22 10:40:01 59


给网友们整理相关的编程文章,网友崔萧玉根据主题投稿了本篇教程内容,涉及到pytorch常用函数、pytorch torch.randn()、pytorch函数、pytorch函数之torch.randn()相关内容,已被763网友关注,下面的电子资料对本篇知识点有更加详尽的解释。

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

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持码农之家。


参考资料

相关文章

  • pytorch中交叉熵损失函数的使用小细节

    发布:2023-04-22

    这篇文章主要介绍了pytorch中交叉熵损失函数的使用细节,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教


  • pytorch中forwod函数在父类中的调用方式解读

    发布:2023-04-06

    这篇文章主要介绍了pytorch中forwod函数在父类中的调用方式解读,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教


  • PyTorch之torch.randn()如何创建正态分布随机数

    发布:2023-04-22

    这篇文章主要介绍了PyTorch之torch.randn()如何创建正态分布随机数问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教


网友讨论