numpy 更改输入的形状

ergxz8rk  于 2023-05-13  发布在  其他
关注(0)|答案(1)|浏览(101)

我有4000个向量。对于每个向量,我们有1个输出值。我想使用ResNet进行回归。现在当输入是这样的:
Tensor([[0.1902,0.5367,0.1589,0.8982,0.7908,0.8838,0.3854],[0.4809,0.7469,0.2444,0.4606,0.6360,0.7149,0.7302],[0.9228,0.1850,0.0877,0.9747,0.5225,0.4538,0.8805]])
在函数resnet(inputs)中,我得到了错误:
RuntimeError:应为conv2d输入3D(未批处理)或4D(批处理),但得到的输入大小为:[3、7]
如何解决这一问题?
我尝试了inputs = inputs.unsqueeze(1).repeat(1,3,1,1),但它不起作用。

nwo49xxi

nwo49xxi1#

import torch

t = torch.tensor(
    [
        [0.1902, 0.5367, 0.1589, 0.8982, 0.7908, 0.8838, 0.3854],
        [0.4809, 0.7469, 0.2444, 0.4606, 0.6360, 0.7149, 0.7302],
        [0.9228, 0.1850, 0.0877, 0.9747, 0.5225, 0.4538, 0.8805],
    ]
)

t.unsqueeze(0)

相关问题