Taichi学习之路（一）

凡尘里的凡人

已于 2022-09-22 16:58:05 修改

阅读量454

点赞数

CC 4.0 BY-SA版权

文章标签：学习 python 开发语言

于 2022-09-22 16:56:53 首次发布

本文链接：https://blog.csdn.net/yinianqi/article/details/117480419

本文介绍了 Taichi Python 库的基本用法，包括 field 的创建与应用、GUI 使用等。field 在 Taichi 中用于表示场或域的概念，可用于定义各种类型的数组结构，如标量场、向量场等，并可通过 kernel 定义并行操作。同时，文章还讨论了 GUI 的简单使用。

1.ti.field()

a global N-d arrary of elements
一般将field理解成是一个场或者域

heat_field = ti.field(dtype = ti.f32, shape = (256,256))

pixel = ti.field(dtype = float, shape=(16,8))
pixel[1,2] = 42.0

vf = ti.Vector.field(3, ti.f32, shape = 4)  # field是4*1的向量场，每一个元素都是3*1的Vector
@ti.kernel
def foo():
	v = ti.Vector([1, 2, 3])
	vf[0] = v

global_scalar = ti.field(dtype = ti.f32, shape=()) # 通过0维的field定义一个全局变量

为了提高运行效率，可以将field改成下面的结构，原理是将运行顺序和存在在内存的顺序调整到一致。
在这里插入图片描述

2.ti.init()

packed的用法，数据结构的shape自动补齐到2的n次幂。

ti.init() # default: packed = False  
a = ti.field(ti.i32,shape=(18,65)) # padded to (32,128) 

ti.init(packed = True)
a = ti.field(ti.i32,shape=(18,65)) # no padding