torch.ones()的简单理解

可keke

已于 2023-12-01 16:04:09 修改

阅读量856

点赞数 7

分类专栏： ML&DL 文章标签： pytorch

于 2023-12-01 15:58:19 首次发布

本文链接：https://blog.csdn.net/qq_44722189/article/details/134736954

版权

ML&DL 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

一、官网链接

torch.ones — PyTorch 2.1 documentation
在这里插入图片描述

二、函数解析

torch.ones(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

返回创建size大小的维度，里面元素全部填充为1

三、举例

a = torch.ones(2, 3) # 相当于a = torch.ones((2, 3))及a = torch.ones(size=(2, 3))
b = torch.ones(5)
c = torch.ones(size=(3,4))
a,b,c

输出：

输出结果如下：
(tensor([[1., 1., 1.],
         [1., 1., 1.]]),
 tensor([1., 1., 1., 1., 1.]),
 tensor([[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]]))

torch.full()以及torch.zeros()类似。

参考：

https://huaweicloud.csdn.net/63807f3cdacf622b8df88edb.html?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Eactivity-1-125103847-blog-103876885.235%5Ev39%5Epc_relevant_anti_t3&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Eactivity-1-125103847-blog-103876885.235%5Ev39%5Epc_relevant_anti_t3&utm_relevant_index=2