mask = torch.zeros_like(logits, requires_grad=True) # 设置 requires_grad=True score = [] for i in range(logits.size(0)): top_100_indices = torch.argsort(logits[i])[-self.topk:] mask_i = mask[i].clone() # 创建副本 mask_i[top_100_indices] = 1.0 score.append(mask_i.clone()) # 创建副本并赋值回 mask return score AttributeError: 'list' object has no attribute 'size'
时间: 2024-04-13 13:29:56 浏览: 149
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
非常抱歉,我之前的回答仍然有误。
这个错误是由于你尝试在一个列表对象上调用`size()`方法,而列表对象并没有`size()`方法。根据你的代码,你想要返回一个包含每个样本的`mask_i`副本的列表,而不是返回一个`score`张量。
以下是修改后的代码示例:
```python
mask = torch.zeros_like(logits, requires_grad=True) # 设置 requires_grad=True
score = []
for i in range(logits.size(0)):
top_100_indices = torch.argsort(logits[i])[-self.topk:]
mask_i = mask[i].clone() # 创建副本
mask_i[top_100_indices] = 1.0
score.append(mask_i.clone()) # 创建副本并添加到 score 列表中
return score
```
在修改后的代码中,我们创建了一个空列表 `score`,然后在循环中使用`append()`方法将每个样本对应的`mask_i`副本添加到列表中。最后返回这个列表作为结果。
希望这次能够解决问题!如果还有其他问题,请随时提问。
阅读全文