torch.cat(-1)
时间: 2023-11-03 17:01:54 浏览: 188
torch-1.0.0
5星 · 资源好评率100%
torch.cat(-1)是一个错误的用法。torch.cat函数的第二个参数应该是一个整数,表示在哪个维度上进行拼接。-1不是一个有效的维度值。正确的用法是根据你的需求选择一个合适的整数作为第二个参数。例如,如果你想在最后一个维度上进行拼接,你可以使用torch.cat(inputs, -1)。这将返回在最后一个维度上拼接后的张量。
阅读全文