美文网首页
pytorch 张量

pytorch 张量

作者: 米线织毛衣 | 来源:发表于2020-02-12 11:40 被阅读0次

    tensor主要有8个属性,如下图。4个与数据相关,4个与梯度求导相关。

    .cuda表示是在gpu上。dtype表示张量的数据类型,一共9种,如下图

    主要是3大类,float型,整型和布尔型。红框框出来的32位float和64位整型用的最多。32位float默认为权重的数据类型,64位整型用作标签的数据类型。

    pytorch中张量的创建方法:

    一、直接创建

    其中pin_memory一般设置为False。

    创建自定义数值张量(数值都相同):

    等差数列的方法创建:

    创建单位阵

    依据概率分布创建:

    创建正态分布:

    创建标准正态分布:

    创建均匀分布

    创建0到n-1的随机排列,用于随机生成索引。还有就是生成伯努利分布

    相关文章

      网友评论

          本文标题:pytorch 张量

          本文链接:https://www.haomeiwen.com/subject/tqiqfhtx.html