tensorflow如何生成规定shape的元素均为1的矩阵
来源:网络收集 点击: 时间:2024-04-24【导读】:
在进行深度学习模型搭建的过程中,经常遇到需要生成和一个tensor拥有相同shape,即结构相同的元素均为1的tensor,用于接下来的运算。本篇经验将介绍如何生成这种tensor,同样地还有元素均为0的tensor。工具/原料morePyCharm方法/步骤1/3分步阅读
2/3
3/3
首先创建一个tensor,规定该tensor的shape为,也就是我们希望生成的元素均为1的矩阵的结构同样为。可以看到此时tensor的结构为2行4列,即shape为

接下来,利用ones_like函数,函数形参为目标tensor。将刚才创建的tensor传给函数,就可以生成相同结构但元素全为1的矩阵了。

同样地,可以利用zeros_like生成结构相同、元素全为0的矩阵。

版权声明:
1、本文系转载,版权归原作者所有,旨在传递信息,不代表看本站的观点和立场。
2、本站仅提供信息发布平台,不承担相关法律责任。
3、若侵犯您的版权或隐私,请联系本站管理员删除。
4、文章链接:http://www.1haoku.cn/art_584072.html
上一篇:3dmax附加后怎么分离
下一篇:深圳新生儿参加医保如何办理