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

如果有一个[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)

这段代码将输出具有相等值的行的下标列表。请注意，这只是一个示例，你可以根据自己的需求进行修改和扩展。