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

标题: CUBLAS是按列进行存储的，吐血。 [打印本页]

作者: xmubingo 时间: 2012-6-16 20:43
标题: CUBLAS是按列进行存储的，吐血。
很简单的例子，比如我们要用一维向量存以下的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

欢迎光临机器学习和生物信息学实验室联盟 (http://123.57.240.48/)