目标需求如下:
有两个tensor的一维向量,需要找到一个张量中的元素在另一个张量中对应的索引。
一个是待找tensor,另一个是目标索引的tensor。
# 待找tensor
label_input = tensor([13, 4, 12, 24, 43, 25, 17, 28, 45, 28, 24, 13, 15, 10, 18, 27, 20, 35,
27, 31,
目标需求如下:
有两个tensor的一维向量,需要找到一个张量中的元素在另一个张量中对应的索引。
一个是待找tensor,另一个是目标索引的tensor。
# 待找tensor
label_input = tensor([13, 4, 12, 24, 43, 25, 17, 28, 45, 28, 24, 13, 15, 10, 18, 27, 20, 35,
27, 31,