机器学习和生物信息学实验室联盟

 找回密码
 注册

QQ登录

只需一步,快速开始

搜索
查看: 2699|回复: 0
打印 上一主题 下一主题

CUBLAS是按列进行存储的,吐血。

[复制链接]
跳转到指定楼层
楼主
发表于 2012-6-16 20:43:52 | 只看该作者 回帖奖励 |倒序浏览 |阅读模式
很简单的例子,比如我们要用一维向量存以下的3*2二维矩阵:
1 2
3 4
5 6
我们用C语言一般是这么存的,按行存储:
int A[] = {1,2,3,4,5,6};

还原成2维矩阵:
你只需要给我A向量,告诉我二维数组是由3行组成,我就明白有6/3=2列,并且按行存储:
→  →
1   2
→  →
3   4
→  →
5   6

如果你要用cublas实现一维向量存放2维矩阵,应该按列存储:
int A[] = {1,3,5,2,4,6};

还原成2维矩阵:
你需要告诉cublas A向量,以及是有3行组成,那么它就明白有2列,并且按列存储:
↓1  ↓2
↓3  ↓4
↓5  ↓6

分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
收藏收藏 转播转播 分享分享
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册

本版积分规则

机器学习和生物信息学实验室联盟  

GMT+8, 2024-11-2 16:30 , Processed in 0.070822 second(s), 32 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表