kernel返回参数如何调用？

综合讨论区
#1 
import taichi as ti
ti.init(arch=ti.gpu,default_fp=ti.f64)
@ti.kernel
def test_kernel() -> ti.Vector:
    m=ti.Vector([1,2,3,4])
    return m
n=test_kernel()
print(n[0])

error：
ValueError: Invalid data type <function Vector at 0x000001906143A0E0>
这返回的向量怎么才能正常调用呢？

#2

Hi @money , ti.kernel 的返回参数可以参考文档： Kernels and functions | Taichi Docs

根据文档，你的代码可以改为：

import taichi as ti
ti.init(arch=ti.cpu)
vec = ti.types.vector(4,ti.f32)
@ti.kernel
def test_kernel() -> vec:
    m=ti.Vector([1,2,3,4])
    return m
n=test_kernel()
print(n[0])