tensor的按维度复制,使用场景:信息检索中,因为query为一条,候选doc为多个,对query和doc算相关性分数的时候复制query。
a = torch.FloatTensor([[1,2]])# shape = [1, 2]
b = a.repeat(2, 4)#分别在各个维度上复制几次
tensor的按维度复制,使用场景:信息检索中,因为query为一条,候选doc为多个,对query和doc算相关性分数的时候复制query。
a = torch.FloatTensor([[1,2]])# shape = [1, 2]
b = a.repeat(2, 4)#分别在各个维度上复制几次