由于找不到taichi内置的排序函数,自己实现了以下的ti.func,但并不能运行
我阅读了 When to use ti.static
with for loops ,有些难以理解,这样的限制虽然提高了性能,但是极大降低了灵活性,连数组排序都不能够完成。
请问有没有可替代的方案 ?
NUM_IMAGES = 3
@ti.func
def sort_small(vector):
j = 0
for i in ti.static(range(1, NUM_IMAGES)):
tmp = vector[i]
j = i
while j >= 1 and tmp<vector[j-1]:
vector[j] = vector[j-1]
j -= 1
vector[j] = tmp
return vector