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)

Jittor 的Op, Var算子相关推荐

Dive into TensorFlow - 解析 TF 核心抽象 op 算子
TF 计算图从逻辑层来讲,由 op 与 tensor 构成.op 是项点代表计算单元,tensor 是边代表 op 之间流动的数据内容,两者配合以数据流图的形式来表达计算图.那么 op 对应的物理层实 ...
今日上午，清华大学发布中国首个高校自研深度学习训练框架—计图Jittor
在百度狂推PaddlePaddle框架,以及旷视计划本月25日发布自研深度学习训练框架MegEngine之际,清华大学突然发布首个中国高校自研深度学习框架Jittor,中文名为计图. 官网链接 htt ...
基于Jittor框架实现LSGAN图像生成对抗网络
基于Jittor框架实现LSGAN图像生成对抗网络生成对抗网络(GAN, Generative Adversarial Networks )是一种深度学习模型,是近年来复杂分布上无监督学习最具前景的 ...
自定义算子高性能开发
自定义算子高性能开发在计图中,一共有三种方法来开发自定义的算子: 使用元算子进行组合. 使用Code算子开发自定义算子. 使用计图编译器编译自定义的模块和custom op. 其中,元算子开发是最为 ...
清华开源 Jittor：首个国内高校自研深度学习框架，一键转换 PyTorch
点击上方"视学算法",马上关注真爱,请设置"星标"或点个"在看" 来源 | 机器之心继 Theano.Caffe 之后,又一个由高校主导 ...
清华开源Jittor：首个国内高校自研深度学习框架，一键转换PyTorch
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! 来源:机器之心@微信公众号继 Theano.Caffe 之后,又一个由高校主导的 ...
添加到收藏夹js(兼容ie/ff/op)不兼容chrome求解决
2019独角兽企业重金招聘Python工程师标准>>> var www_yibo_com = function (obj, url, title) {var e = window.e ...
OneFlow源码解析：Op、Kernel与解释器
撰文|郑建华更新|赵露阳 1 Op与Kernel的注册继续追踪执行流程会发现,ReluFunctor在构造UserOpExpr时会用到UserOpRegistryMgr管理的Op与Kernel. ...
javascript算法+手写js面试题
链表 function ListNode(val, next) {this.val = (val===undefined ? 0 : val)this.next = (next===undefined ...

Jittor 的Op, Var算子

Output: float32[3,]

Output: [ 1. 2. 3.]

Jittor 的Op, Var算子相关推荐

最新文章

热门文章