Jittor 的Op, Var算子
要使用jittor訓練模型，需要了解兩個主要概念：
? Var：Jittor的基本數(shù)據(jù)類型
? Operations：Jittor的算子與numpy類似
首先，開始使用Var。Var是jittor的基本數(shù)據(jù)類型，為了運算更加高效Jittor中的計算過程是異步的。如果要訪問數(shù)據(jù)，可以使用Var.data進行同步數(shù)據(jù)訪問。
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創(chuàng)建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)

總結(jié)

以上是生活随笔為你收集整理的Jittor 的Op, Var算子的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。