deep learning with pytorch——4

大小、存儲偏移量和跨步數 1.爲了索引到存儲中,張量依賴於一些信息,這些信息連同它們的存儲一起,明確地定義了它們:大小、存儲偏移量和跨距(圖2.5)。size(或者shape,用NumPy術語來說)是一個元組,表示張量的每個維度上有多少個元素。存儲偏移量是存儲中對應於張量中第一個元素的索引。跨距是存儲中需要跳過的元素數,以便沿着每個維度獲取下一個元素。 2.通過提供相應的索引,可以獲得張量中的第二
相關文章
相關標籤/搜索