如果有[10,1]张量，如何获得那些具有相等值及其下标的行？

import torch

# 创建一个[10,1]张量
tensor = torch.tensor([[1], [2], [3], [2], [1], [4], [3], [2], [1], [4]])

# 创建一个空列表，用于存储具有相等值的行的下标
equal_rows = []

# 遍历张量的每一行，并记录具有相等值的行的下标
for i, row in enumerate(tensor):
    # 获取当前行的值
    value = row.item()
    
    # 检查当前行的值是否与其他行的值相等
    if (tensor == value).any():
        # 将当前行的下标添加到列表中
        equal_rows.append(i)

# 打印具有相等值的行的下标
print(equal_rows)