tensor={16,20,1,64,64},写一段pytorch代码,实现该tensor中dim=1的逐个相减。
时间: 2023-03-27 21:00:18 浏览: 118
pytorch 改变tensor尺寸的实现
可以使用PyTorch中的torch.unbind()函数来实现该tensor中dim=1的逐个相减,代码如下:
```
import torch
tensor = torch.randn(16, 20, 1, 64, 64)
tensor_list = torch.unbind(tensor, dim=1)
result = tensor_list[]
for i in range(1, len(tensor_list)):
result = result - tensor_list[i]
print(result.shape)
```
这段代码先使用torch.unbind()函数将tensor在dim=1的维度上拆分成多个tensor,然后逐个相减,最终得到结果。
阅读全文