torch.expand()
Posted junzhaoliang
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch.expand()相关的知识，希望对你有一定的参考价值。
a=torch.rand(32,1)
a.expend(imgs.size())
>>>RuntimeError: The expanded size of the tensor (256) must match the existing size (32) at non-singleton dimension 2. Target sizes: [32, 1, 256, 170]. Tensor sizes: [32, 1]
a=torch.rand(32,1,1,1)
a.expend(imgs.size())
print(a.size())
>>>(32,1,256,170)
以上是关于torch.expand()的主要内容，如果未能解决你的问题，请参考以下文章