Torch7深度学习教程（三）

/article/11003146.html

转载请注明出处

函数的使用



这是函数的定义方式，声明的关键字+定义的函数名+形参的名字，在此博主返回两个值，具体的函数功能在后面再说



这是初始化一个5x2的矩阵，并且初值都为1。这里有多了一种初始化矩阵的方法。



这是先声明一个2x5的矩阵，然后再调用fill（）方法其值全部初始化为4。



将a，b矩阵输入到addTensors函数里面，注意这里是实参，前面定义的a，b是形参，这个有点c基础的应该都分的清楚吧。打印结果就是返回a矩阵和axb的矩阵。



这样子写更像matlab，返回的a和axb分别按顺序赋值给c，d。

CUDA的Tensor

抱歉了，我的电脑不是NVIDIA的显卡，这里以后再补上吧，看不到运行结果



这里就是多调用一下cuda（）函数，然后计算axb矩阵的乘机运算就可以了。开头的require ‘cutorch’就是导入cuda运算包，类似于c语言里面的include。

下一章，开始进入主题——神经网络