可否新增Tensor的操作,如topk()升序取前几项,tensor根据维度和索引赋值等?
Created by: qianledan
对比了paddle和pytorch关于tensor的相关操作,本人需要用到“tensor的升序取最小的前面几项”处理,和根据索引赋值操作,但是paddle里没有相关的函数,pytorch里有,如:topk(k, dim=None, largest=True, sorted=True),正反排序都可以。index_fill(dim, index, value)依据维度和索引赋值,index_select(dim, index)根据维度和索引选择。
pytorch里关于tensor的操作函数非常全面,而paddle里却不多,只有少数。希望工程师可以参照pytorch里的函数进行新增。