pytorch学习杂记

1.torch.cat([a,b],0)

对应numpy的 np.concatenate((a, b), axis=0) 函数，在第一个维度拼接两个矩阵。

2.

 

转载于:https://www.cnblogs.com/leokale-zz/p/11599164.html