RuntimeError: The expanded size of the tensor (514) must match the existing size (512) at non-singleton dimension 1. Target sizes: [8, 514]. Tensor sizes: [1, 512]

RuntimeError: The expanded size of the tensor (514) must match the existing size (512) at non-singleton dimension 1. Target sizes: [8, 514]. Tensor sizes: [1, 512] #708