|
|
之前我们介绍太极语言的ti.kernel内核的时候，提到它支持的参数类型，其中举例所i32和f32这两种，这是太极编程语言中的数据类型。机智客觉得在编程语言中，数据类型广泛用于函数和方法的参数中，比如在kernel中，比如在ti.field方法中，所以是基础，重要性毋庸置疑。, w/ W( M, M6 o. Q/ f: x$ ~
, l" {& K1 p1 v3 r/ P% L
和其他编程语言类似，太极目前的数据类型也有几种：有符号整数类型，无符号整数类型，浮点类型。太极不同的数据都是由数据类型和精度组成的。其中，Signed integers包含ti.i8，ti.i16，ti.i32，ti.i64。Unsigned integers类型则用u为前缀，比如ti.u8，ti.u16等。而浮点则有两种，ti.f32和ti.f64。机智客在以前的文章里提到的i32和f32，其实是太极语言里用得最普遍的。当然和其他编程语言不太一样的也有，太极语言的数据类型并不多，布尔类型则是由ti.i32表示。
0 G% j S, A7 P7 ]
! D9 V' y1 P2 X$ D+ B1 q5 s1 j 默认的数据类型精度都是32位。需要说明的是，CPU和CUDA后端支持所有的数据类型，而其他则由于后端API的约束，要参考相应后端的API才能确定是否缺少某些数据类型支持。" w3 T0 g0 |- }- h
. r6 H0 b2 d0 R0 r) p) t# e0 ]
上面是太极语言的数据类型，可用于函数里的参数。太极语言规定只能是标量，返回也是。如果函数有返回值，则要指定返回数据类型。而且返回值也只能是一个标量（不能是两个以上），而不能是矢量或矩阵，哪怕Python中的元组也不行。所以参数可以有多个标量，而返回只能是一个标量。比如这段来源于英文文档中的代码片段。8 C# y" `0 B% l# N! Y. Q* |( L
( \9 B( b* s3 }2 \ j+ t5 M4 b. |% | ti.kernel. W0 Q, X$ q1 S, U0 ?# o
" z7 ? J+ t) U2 t! Q) k+ H
def my_kernel()->ti.f32:
, u) T' X* J! [" Z% v/ v" s
! `; r& Y: C! g. c3 _9 T return 233.33
! Z) [9 {& [9 l
1 t' U; y1 ?2 r% h% a 我们以前介绍过，太极语言的初始化，说过，初始化，也可以指定数据类型。这里说下，如果我们要设置初始化的数据类型，可以这样ti.init(default_ip=ti.i64,default_fp=ti.f32)。当然了，数据类型这块，还有很多其他细节知识，比如数据类型转化，矢量和矩阵的知识等。我们以后遇到再学习。
! x5 D: ^( A7 E0 K2 G E3 F# |# V3 ~3 \4 o/ {; F0 i: B$ i6 ^4 {
taichi而关于太极语言的作用域，相信学习了上文或者以前文章的朋友，都了解了。其实就是 ti.kernel或 ti.func装饰的代码区域内。ti.kernel里的是太极内核，ti.func里的是太极函数。可以调用ti.func里的函数，也可以嵌套调用函数，然而不支持递归函数。外部就是普通Python代码了，不同代码不同处理，各找各妈，各归各管。如果用CUDA做类比的话，Taichi作用域等于device side。taichi https://taichi-lang.cn/9 b i \* \% s
9 O/ p; j3 |, s$ ^ `
) B: H! i- i* f k
|
|
