Theano学习之基本用法

1、函数定义,容器定义,原始方程打印

import numpy as np #需要用到的模块
import  theano.tensor as T #用来定义张量,容器等
from  theano import function#用来定义函数
from  theano import pp#可以用来打印原始方程

x = T.dscalar('x')#建立x的容器
y = T.dscalar('y')#建立y的容器
z = x+y#建立方程

f = function([x,y],z)#使用function定义theano方程,然后将值x,y输入[]中,输出值放在后面z 里
print(f(2,3))
print(pp(z))

结果:

2、矩阵定义

import numpy as np #需要用到的模块
import  theano.tensor as T #用来定义张量,容器等
from  theano import function#用来定义函数
from  theano import pp

x = T.dmatrix('x')#建立矩阵x的容器
y = T.dmatrix('y')#建立矩阵y的容器
z = x+y#建立方程
f = function([x,y],z)#使用function定义theano方程,然后将值x,y输入[]中,输出值放在后面z 里

print(pp(z))
print(f(
        np.arange(12).reshape((3,4)),
        10*np.ones((3,4))
        )
      )#如果在定义容器时是定义的矩阵,那么函数里面放的也要是矩阵形式,这里两个矩阵相加

结果:

来源

猜你喜欢

转载自blog.csdn.net/weixin_40849273/article/details/84566745