如果有一个[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)
这段代码将输出具有相等值的行的下标列表。请注意,这只是一个示例,你可以根据自己的需求进行修改和扩展。
领取专属 10元无门槛券
手把手带您无忧上云