RuntimeError with copy_() and expand() - Unsupported operation: more than one element of the written

报错代码：

feature = feature.unsqueeze(2).expand(B, K, F) # squeeze(n)的参数n指定新维度的位置，expand():返回当前张量在某维扩展更大后的张量，扩展的维度初始必须是1维的
feature_bank = feature_bank.unsqueeze(0).expand(B, K