RuntimeError: Sizes of tensors must match except in dimension 0. Expected size 96 but got size 1 for tensor number 1 in the list.
时间: 2023-11-06 07:57:00 浏览: 179
pytorch dataloader 取batch_size时候出现bug的解决方式
这个错误通常出现在 PyTorch 运行时,它表示两个张量在除了第0个维度之外的维度上的大小不匹配。具体来说,这个错误信息中的 "Expected size 96 but got size 1 for tensor number 1 in the list" 表示第一个张量的形状为 `(96, ...)`, 而第二个张量的形状为 `(1, ...)`,其中 `...` 表示其他维度的形状。
这个错误通常是由于在进行张量拼接、广播或者其他操作时,两个张量的形状不兼容引起的。可以检查一下这两个张量的形状,看看它们在非第0个维度上的大小是否一致,如果不一致的话,可以考虑使用 reshape 或者其他操作来调整形状,以使得它们在相应维度上的大小一致。
阅读全文