我要用pytorch对比两个张量中的元素,选出较小的一个组合成一个新的张量
时间: 2023-06-07 22:09:19 浏览: 86
用于张量分解网络的Pytorch工具包
5星 · 资源好评率100%
可以使用pytorch的min函数,将两个张量作为参数传入即可。示例代码如下:
```
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([2, 1, 4])
c = torch.min(a, b)
print(c)
```
输出结果为:
```
tensor([1, 1, 3])
```
阅读全文