PyTorch Note -- 合并与分割、数学运算、属性统计

stack 会创建新的维度

矩阵乘法

统计属性

where用法

设计where的初衷：如果采用循环控制语句for i in range(k)，for j in range(m)等，这种方法是不能用GPU来加速的。所以设计了where方法，condition矩阵可以用cpu的方式或者gpu的方式生成，一但生成之后，就可以用where，进而使用GPU来加速了。