网站首页 网站建设 IT知识 IT知识教程 pytorch中tensor.expand()和tensor.expand_as()函数详解

pytorch中tensor.expand()和tensor.expand_as()函数详解

2021-05-22 21:31:24

tensor.expend()函数

>>> import torch>>> a=torch.tensor([[2],[3],[4]])>>> print(a.size())torch.Size([3, 1])>>> a.expand(3,2)tensor([[2, 2], [3, 3], [4, 4]])>>> atensor([[2], [3], [4]])

可以看出expand()函数括号里面为变形后的size大小,而且原来的tensor和tensor.expand()是不共享内存的。

tensor.expand_as()函数

>>> b=torch.tensor([[2,2],[3,3],[5,5]])>>> print(b.size())torch.Size([3, 2])>>> a.expand_as(b)tensor([[2, 2], [3, 3], [4, 4]])>>> atensor([[2], [3], [4]])

可以看出,b和a.expand_as(b)的size是一样大的。且是不共享内存的。

以上这篇pytorch中tensor.expand()和tensor.expand_as()函数详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。 如涉及版权问题,请提交至online#300.cn邮箱联系删除。

以上产品还未完全满足我的所有需求,在下方提交我的专属需求
我的专属需求:
*手机号:
*验证码:
img
咨询报价
现在咨询
img

在线咨询

建站在线咨询

img

微信咨询

扫一扫添加
动力姐姐微信

img
img

TOP