Torch7深度学习教程(三)

  • 函数的使用
    这里写图片描述
    这是函数的定义方式,声明的关键字+定义的函数名+形参的名字,在此博主返回两个值,具体的函数功能在后面再说
    Torch7深度学习教程(三)_第1张图片
    这是初始化一个5x2的矩阵,并且初值都为1。这里有多了一种初始化矩阵的方法。
    Torch7深度学习教程(三)_第2张图片
    这是先声明一个2x5的矩阵,然后再调用fill()方法其值全部初始化为4。
    Torch7深度学习教程(三)_第3张图片
    将a,b矩阵输入到addTensors函数里面,注意这里是实参,前面定义的a,b是形参,这个有点c基础的应该都分的清楚吧。打印结果就是返回a矩阵和axb的矩阵。
    Torch7深度学习教程(三)_第4张图片
    这样子写更像matlab,返回的a和axb分别按顺序赋值给c,d。

  • CUDA的Tensor

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

Torch7深度学习教程(三)_第5张图片
这里就是多调用一下cuda()函数,然后计算axb矩阵的乘机运算就可以了。开头的require ‘cutorch’就是导入cuda运算包,类似于c语言里面的include。

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

你可能感兴趣的:(torch)