其实和Numpy的广播是差不多的,就是改变小矩阵的形状为大矩阵的形状然后运算。具体参考:使用Pytorch广播语义实现不同形状数组的运算 - pytorch中文网
原文出处: https://ptorch.com/news/39.html
转载于:https://my.oschina.net/earnp/blog/1511615
其实和Numpy的广播是差不多的,就是改变小矩阵的形状为大矩阵的形状然后运算。具体参考:使用Pytorch广播语义实现不同形状数组的运算 - pytorch中文网
原文出处: https://ptorch.com/news/39.html
转载于:https://my.oschina.net/earnp/blog/1511615