在函数后面写->ti.Vector()似乎并不行
Hi @shiro, 抱歉我们的文档还在持续完善中。
关于kernel能返回 ti.Vector
和 ti.Matrix
, 我写了一个小例子你可以参考:
import taichi as ti
ti.init()
A = ti.Vector.field(3, ti.f32, ())
B = ti.Matrix.field(2,2, ti.i32, ())
@ti.kernel
def test_vector()-> ti.types.vector(3,ti.f32):
return ti.Vector([1.0, 3.0, 4.0])
@ti.kernel
def test_matrix() ->ti.types.matrix(2,2,ti.i32):
return ti.Matrix([[1,2],[3,4]])
A[None] = test_vector()
print(A[None])
B[None] = test_matrix()
print(B[None])
如果想要了解更多细节可以参考相关的PR:here。
不太理解ti.Vector和ti.field的区别,看文档发现基本都使用field代替了ti.Vector是为什么?