1. 程式人生 > >Theano學習之基本用法

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))
        )
      )#如果在定義容器時是定義的矩陣,那麼函式裡面放的也要是矩陣形式,這裡兩個矩陣相加

結果:

來源