Pytorch的Tensor操作(2)

  Broadcasting 依舊是擴展功能,且不需要拷貝數據 過程: 先檢查維度是否相同 從最小維度(最右面最小)開始匹配,在前面添加1維度 然後將維度爲「1」的部分,擴展到對應維度相同大小 實際應用,對某個Tensor加一個標量,將標量自動匹配到目標張量的數據類型 適用情況: (1): 某一維度相同,或爲1 (2):某一部分維度相同 (3):不符合規則:只能爲「1」或相同 拼接與拆分 Cat
相關文章
相關標籤/搜索