给一个n*1的Tensor 返回相同元素的值以及数量
时间: 2024-11-03 11:18:37 浏览: 13
tensor:包张量在Go中提供了有效且通用的n维数组,可用于机器学习和深度学习
在Python的PyTorch库中,如果你有一个形状为`(n, 1)`的张量(tensor),它实际上是一个一维数组,你可以使用`torch.unique`函数来获取所有唯一的元素,并同时计算每个独特元素的数量。这个函数会返回一个元组,第一个元素是包含唯一元素的新张量,第二个元素是一个整数张量,表示每个独特元素在原始张量中出现的次数。
例如:
```python
import torch
# 假设我们有这样一个张量
input_tensor = torch.tensor([1, 2, 2, 3, 4, 4, 4])
# 使用torch.unique计算唯一元素及其计数
unique_values, counts = torch.unique(input_tensor, return_counts=True)
print("Unique values:", unique_values)
print("Counts:", counts)
```
输出将会像这样:
```
Unique values: tensor([1, 2, 3, 4])
Counts: tensor([1, 2, 1, 3])
```
阅读全文