我想在简单的@ti.func中创建不同长度的向量,但总是不成功。
import taichi as ti
ti.init(arch=ti.cpu, cpu_max_num_threads=1)
@ti.func
def Pick_NodeID( NodeNumber):
NodeID = ti.Vector.zero(float, NodeNumber)
return NodeID
@ti.kernel
def text():
NodeNumber = 8
NodeID = Pick_NodeID(NodeNumber)
text()