在您提供的代码中,outputs 的输出确实是一个张量,但是这并不代表 return_all_layers 参数没有生效。如果您将 return_all_layers 设置为 False,模型只会返回最后一个时间步的输出(即最后一个隐藏状态),而不是返回所有时间步的输出。因此,outputs 张量的形状应该为 [batch_size, num_hidden_units, height, width],其中 batch_size 是当前批次中的样本数量,num_hidden_units 是隐藏单元的数量,height 和 width 是图像的高度和宽度。在您的代码中,outputs 张量的形状符合这个规律,因此可以确认 return_all_layers 参数已经生效。

另外,关于您的代码中的一个小问题:在 forward 函数中,您定义了一个变量 c_cur,但是并没有使用它。这个变量应该是多余的,可以直接删除。

这是我按照你提供的建议改的代码我的return_all_layers 设置为 False但是我的输出outputs仍然是tensor-00146 -00113 -00137 00040 00058 00087 -00146 -00113 -00137 00040 00058 00087 -00174 -00124 -00138 -

原文地址: https://www.cveoy.top/t/topic/brGN 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录