torch.repeat与numpy中的np.tile()类似,可以视作将tensor作为一个元素,按照指定倍数进行复制,维度从后往前匹配,输入维度要小于指定的维度,tensor.repeat(2,1,1)= tensor.repeat((2,1,1))
pytorch学习笔记——torch.repeat(),torch.expend()
于 2022-06-13 02:22:27 首次发布
torch.repeat与numpy中的np.tile()类似,可以视作将tensor作为一个元素,按照指定倍数进行复制,维度从后往前匹配,输入维度要小于指定的维度,tensor.repeat(2,1,1)= tensor.repeat((2,1,1))