美文网首页
【函数学习】torch—tensor张量生成

【函数学习】torch—tensor张量生成

作者: 风萧萧兮水易寒 | 来源:发表于2020-02-14 17:21 被阅读0次

1.torch.linspace(start, end, num=100, out=None) 

    返回一个1维张量,包含在区间start 和 end 上均匀间隔的num个点。 输出1维张量的长度为num

    参数:

        start (float) – 序列的起始点

        end (float) – 序列的最终值

        num (int) – 在start 和 end间生成的样本数

        out (Tensor, optional) – 结果张量

2.torch.eye(n, m=None, out=None)

    返回一个2维张量,对角线位置全1,其它位置全0

    参数:

        n (int ) – 行数

        m (intoptional) – 列数.如果为None,则默认为n

        out (Tensoroptinal) - Output tensor

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

    返回一个张量,包含了从区间[0,1)的均匀分布中抽取的一组随机数,形状由可变参数sizes 定义

    参数:

        sizes (int...) – 整数序列,定义了输出形状

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

    返回一个张量,包含了从标准正态分布(均值为0,方差为 1,即高斯白噪声)中抽取一组随机数,形状由可变参数sizes定义

    参数:

        sizes (int...) – 整数序列,定义了输出形状

5.torch.arange(start, end, step=1, out=None) → Tensor

    返回一个1维张量,包含从start到end,以step为步长的一组序列值(默认步长为1)。

    参数:

        start (float) – 序列的起始点

        end (float) – 序列的终止点

        step (float) – 步长,相邻点的间隔大小

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

    返回一个全为标量 0 的张量,形状由可变参数sizes 定义

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

    返回一个全为1 的张量,形状由可变参数sizes定义。

参考与详细:

官方文档  https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/#creation-ops

感谢!

相关文章

网友评论

      本文标题:【函数学习】torch—tensor张量生成

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