Jittor 的Op, Var算子

Jittor 的Op, Var算子

要使用jittor训练模型，需要了解两个主要概念：

• Var：Jittor的基本数据类型
• Operations：Jittor的算子与numpy类似

首先，开始使用Var。Var是jittor的基本数据类型，为了运算更加高效Jittor中的计算过程是异步的。如果要访问数据，可以使用Var.data进行同步数据访问。

import jittor as jt

a = jt.float32([1,2,3])

print (a)

print (a.data)

# Output: float32[3,]

# Output: [ 1. 2. 3.]

Jittor的算子与numpy类似。尝试一些操作，通过操作jt.float32创建Var a和b，并相加。输出这些变量相关信息，可以看出具有相同的形状和类型。

import jittor as jt

a = jt.float32([1,2,3])

b = jt.float32([4,5,6])

c = a+b

print(a,b,c)

除此之外，使用的所有算子jt.xxx(Var,...)都具有别名Var.xxx(...)。 例如：

c.max() # alias of jt.max(a)

c.add(a) # alias of jt.add(c, a)

c.min(keepdims=True) # alias of jt.min(c, keepdims=True)

如果想知道Jittor支持的所有操作，可以运行help(jt.ops)。 在jt.ops.xxx中找到的所有操作，都可以通过别名jt.xxx。

help(jt.ops)