output_img1 = torch.from_numpy(output_img1.copy()).float() AttributeError: 'Tensor' object has no attribute 'copy'
时间: 2023-10-16 13:32:37 浏览: 129
one hot编码:`torch.Tensor.scatter_()`函数用法详解
这个错误提示是因为 `Tensor` 对象没有 `copy` 方法。如果你想将一个 Tensor 转换为 NumPy 数组并进行 copy 操作,可以使用 `numpy()` 方法进行转换,然后使用 `copy()` 方法进行复制,例如:
```python
output_img1 = output_img1.cpu().numpy().copy()
output_img1 = torch.from_numpy(output_img1).float()
```
这里的 `cpu()` 方法是将 Tensor 从 GPU 移回 CPU,如果你的 Tensor 已经在 CPU 上,可以省略这一步。另外,`numpy()` 方法将 Tensor 转换为 NumPy 数组,这样就可以使用 `copy()` 方法进行复制了。最后,再将 NumPy 数组转换为 Tensor 对象。
阅读全文