- 函数的使用
这是函数的定义方式,声明的关键字+定义的函数名+形参的名字,在此博主返回两个值,具体的函数功能在后面再说
这是初始化一个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。
下一章,开始进入主题——神经网络
时间: 2024-10-27 04:18:49