躺不平的废柴 2023-04-07 10:43 采纳率: 75%
浏览 281
已结题

遇到RuntimeError: shape '[1, 1]' is invalid for input of size 4

RuntimeError: shape '[1, 1]' is invalid for input of size 4

报错部分
batch_indices = torch.arange(B, dtype=torch.long).to(device).view(view_shape).repeat(repeat_shape)

B为4
view_shape为[1, 1]

请问应该如何解决呢?

  • 写回答

1条回答 默认 最新

  • isMae 2023-04-07 11:16
    关注

    这样试试呢?

    batch_indices = torch.arange(B, dtype=torch.long).to(device).view(B, 1).repeat(repeat_shape)
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 4月15日
  • 已采纳回答 4月7日
  • 创建了问题 4月7日