torch.expand()

admin 2023-05-19 09:54:02

篇首语：本文由小编为大家整理，主要介绍了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()的主要内容，如果未能解决你的问题，请参考以下文章

如何在library中使用productFlavors

android中Fragment的切换方法。

torch.expand()

您可能还会对下面的文章感兴趣：

相关文章