satuma77 https://blog.hatena.ne.jp/satuma77/ satuma-portfolio https://satuma-portfolio.hateblo.jp/ メモ WIP メモ．後でちゃんと書く 結論：次が最速 cupy.mean(cupy.stack(cupy.split(data, split_size)), axis=axis) ダメなケース：cupy.split(data, split_size)がリストで返ってくるからといって，cupy.array()でcupy配列にしてから平均を取ろうとすると，GPUとの無駄な通信が発生して，実行速度が遅くなる．GPU・CPU間の通信は極力減らすのが鉄則． cupy.mean(cupy.array(cupy.split(data, split_size), dtype='f'), axis=axis) # かなり遅い… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Fsatuma-portfolio.hateblo.jp%2Fentry%2F2018%2F01%2F26%2F101327" title=" CuPyで配列を部分配列に切り分けて結合してから平均 - satuma-portfolio" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2018-01-26 10:13:27 rich https://satuma-portfolio.hateblo.jp/entry/2018/01/26/101327 1.0 100%