从 Pytorch tensor 存储空间的连续性 (contiguous) 说到 4D tensor 的存储格式 (memory_format)
起源:令人一头雾水的 contiguous刚开始阅读 Pytorch 代码的时候,碰到别的作者处理 tensor,有时候会在后面加上 contiguous,就觉得十分奇怪,不明白 contiguous 的含义,看了相关的解释 之后,好像理解了一点,就这浅薄的理解做点笔记,总结一下。概念解释:什么是 contiguous如果让我用一句话概括 tensor 的 contiguous 特性,我会说:“tensor 的元素按维度储存在连续的内存地址中”,什么是按维度呢?以 (H, W) 的2维 tensor
原创
2020-09-22 16:26:09 ·
3177 阅读 ·
2 评论