Memory layout

PyTorch Developer Podcast

コンテンツは PyTorch, Edward Yang, and Team PyTorch によって提供されます。エピソード、グラフィック、ポッドキャストの説明を含むすべてのポッドキャストコンテンツは、PyTorch, Edward Yang, and Team PyTorch またはそのポッドキャストプラットフォームパートナーによって直接アップロードされ、提供されます。誰かがあなたの著作物をあなたの許可なく使用していると思われる場合は、ここで概説されているプロセスに従うことができますhttps://ja.player.fm/legal。

4+ y ago 16:26

MP3•エピソードのホーム

Memory layout specifies how the logical multi-dimensional tensor maps its elements onto physical linear memory. Some layouts admit more efficient implementations, e.g., NCHW versus NHWC. Memory layout makes use of striding to allow users to conveniently represent their tensors with different physical layouts without having to explicitly tell every operator what to do.

Further reading.

Tutorial https://pytorch.org/tutorials/intermediate/memory_format_tutorial.html
Memory format RFC https://github.com/pytorch/pytorch/issues/19092
Layout permutation proposal (not implemented) https://github.com/pytorch/pytorch/issues/32078

83 つのエピソード