想知道这么把taichi的数据类型转化为tensor类型,想用pytorch处理taichi的输出数据
您好,请参阅文档的 与外部数组进行交互 页面:
@ti.kernel
def my_kernel():
for i in x:
x[i] = i * 2
x = ti.field(ti.f32, 4)
my_kernel()
x_torch = x.to_torch()
print(x_torch) # torch.tensor([0, 2, 4, 6])
x.from_numpy(torch.tensor([1, 7, 3, 5]))
print(x[0]) # 1
print(x[1]) # 7
print(x[2]) # 3
print(x[3]) # 5
这文档搜索里to_torch()搜不到
可以把滚动条往下拉就会显示更多结果~