RuntimeError: Expected 4-dimensional input for 4-dimensional weight [16, 1, 3, 3], but got 3-dimensi
可以看到这句提示,大致就是我们传入的数据输入到CNN网络,然后由于维度不同导致的。因为我们输入的是四维,但是得到的却是三维。使用SAC训练机器人导航至目标点时报错,网络使用的是自定义的CNN,然后在运行的时候出现了如标题所示的这种小错误。在PyTorch中,对于卷积神经网络(CNN),通常期望的输入是一个四维的张量(Tensor),其形状为。维度应该是1,但是你需要确保这个维度在传递给卷积层之前被正确地包含在输入张量中。forward函数的第一行显示有报错,报错内容如下。,那么它缺少了一个维度,即。
原创
2024-05-31 17:35:52 ·
259 阅读 ·
0 评论