在PyTorch的张量中将一个数字范围内插到另一个

时间:2019-12-23 02:01:57

标签: python pytorch tensor

我正在尝试将一组数字,例如{0,4,6}转换为连续的{0,1,2}。 目前,我正在使用Scipy的{​​{1}}函数。我的问题是我想更改张量的每个元素。有没有比向张量中的每个元素广播interp1d函数更好的方法了?如果没有,我该如何实现自定义功能的广播

当前我正在执行以下操作:

interp1d

1 个答案:

答案 0 :(得分:1)

如果只想替换,为什么要插值?

a[a==6]=2
a[a==4]=1

应该工作。如果您不想进行硬编码,则可以像查找一样找到唯一值,对它们进行排序,然后循环替换它们。