参考:菜鸟教程
整数数组索引,可以将矩阵的部分或全部元素按任意顺序组合为一个新矩阵,原矩阵中的元素在新矩阵中可出现0次或多次,新矩阵可以为任意大小
下面以numpy举例,pytorch 类似
例1:
其中row的shape[0]决定输出矩阵的行数,col的shape[1]决定输出矩阵的列数
例2:(接上例)
broadcast机制:row的shape[1]为1,或col的shape[0]为1时可进行broadcast,否则row和col的shape必须相同
参考:菜鸟教程
整数数组索引,可以将矩阵的部分或全部元素按任意顺序组合为一个新矩阵,原矩阵中的元素在新矩阵中可出现0次或多次,新矩阵可以为任意大小
下面以numpy举例,pytorch 类似
例1:
其中row的shape[0]决定输出矩阵的行数,col的shape[1]决定输出矩阵的列数
例2:(接上例)
broadcast机制:row的shape[1]为1,或col的shape[0]为1时可进行broadcast,否则row和col的shape必须相同