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

Tensorflow 我可以将压缩数据传输到GPU,然后在GPU上解压缩吗?,tensorflow,Tensorflow,我试图将大型稀疏矩阵发送到GPU进行计算,大部分计算时间都花在将数据从CPU传输到GPU上。然而,我的数据实际上非常稀疏(很多零值),所以我想在传输前对其进行大幅压缩,以帮助缓解带宽瓶颈。这在Tensorflow中可行吗?当然。SparseTensor类或cuSPARSE库是否满足您的需要?当然可以转换为COO、CSR或CSC等格式。SparseTensor是库的一部分,并有文档记录,但不幸的是,我正在路上,现在无法测试。

我试图将大型稀疏矩阵发送到GPU进行计算,大部分计算时间都花在将数据从CPU传输到GPU上。然而,我的数据实际上非常稀疏(很多零值),所以我想在传输前对其进行大幅压缩,以帮助缓解带宽瓶颈。这在Tensorflow中可行吗?

当然。SparseTensor类或cuSPARSE库是否满足您的需要?当然可以转换为COO、CSR或CSC等格式。

SparseTensor是库的一部分,并有文档记录,但不幸的是,我正在路上,现在无法测试。