c = torch.tensor([[1, 2], [3, 4]]) print(torch.max(c, dim=0)) print(torch.min(a, dim=0))
时间: 2024-05-03 11:17:57 浏览: 108
用c语言实现最大最小值
这段代码存在问题,因为变量 a 没有定义。假设你想使用变量 c 进行计算,那么以下是正确的代码:
```python
import torch
c = torch.tensor([[1, 2], [3, 4]])
print(torch.max(c, dim=0))
print(torch.min(c, dim=0))
```
这段代码的作用是找到张量 c 在每一列上的最大值和最小值。`torch.max(c, dim=0)` 返回一个元组,包含两个张量,第一个张量是每一列上的最大值,第二个张量是每个最大值所在的行索引。`torch.min(c, dim=0)` 的返回值同理,只不过是每一列上的最小值和所在行索引。
阅读全文