我可以将压缩数据传输到GPU然后在GPU上解压缩吗?

时间:2016-09-07 04:51:12

标签: tensorflow

我正在尝试向GPU发送用于计算的大型稀疏矩阵,并且大部分计算时间用于将数据从CPU传输到GPU。然而,我的数据实际上非常稀疏(许多零值),所以我有一个想法是在预转移时显着压缩它以帮助减轻带宽瓶颈。这在Tensorflow中是否可行?

1 个答案:

答案 0 :(得分:0)

不确定。 SparseTensor类或cuSPARSE库是否满足您的需求?绝对可以转换为COO,CSR或CSC等格式。