Pytorch的Tensor操作(2)

  Broadcasting 依旧是扩展功能,且不需要拷贝数据 过程: 先检查维度是否相同 从最小维度(最右面最小)开始匹配,在前面添加1维度 然后将维度为“1”的部分,扩展到对应维度相同大小 实际应用,对某个Tensor加一个标量,将标量自动匹配到目标张量的数据类型 适用情况: (1): 某一维度相同,或为1 (2):某一部分维度相同 (3):不符合规则:只能为“1”或相同 拼接与拆分 Cat
相关文章
相关标签/搜索