2016-09-07 16 views
2

GPUに大規模な疎行列を送信しようとしていますが、計算時間の大半はCPUからGPUにデータを転送するのに費やされます。しかし、私のデータは実際には非常にまばらです(ゼロの値が非常に多い)ので、帯域幅の狭いボトルネックを緩和するためにデータを大幅に圧縮するという考えがありました。これはTensorflowで実現可能ですか?圧縮データをGPUに転送してから、GPUで解凍することはできますか?

答えて

0

SparseTensorクラスまたはcuSPARSEライブラリはニーズを満たしていますか? COO、CSR、CSCなどの形式に変換することは間違いありません。

+0

TFでこれを行う方法についてはほとんど分かっていません... – jstaker7

+0

SparseTensorはライブラリの一部であり、文書化されていますが、残念ながら私は道に迷い込んでいます。 – Davislor

関連する問題