torch入门笔记9:function


  • 函数的使用 
    这里写图片描述 
    这是函数的定义方式,声明的关键字+定义的函数名+形参的名字,在此博主返回两个值,具体的函数功能在后面再说 
    这里写图片描述 
    这是初始化一个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。

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值