标签:数据 size sp ad res c bs 空间 stl
矩阵维度表示和正常相反。 a[2][3],行2列3,a.shape.shape_[0]=3,a.shape.shape_[1]=2.
pred.Resize( Shape2( batch_size, num_out ) );
Resize相当于STL的reserve,预先分配空间。Shape2说明是一个二维的矩阵。
Slice 从高维分割矩阵(tensor),得到的仍然是矩阵(tensor)形式。
Copy( ninput, inbatch ); 可以作GPU,CPU 数据传输。
dot(A,B)=B^T*A
标签:数据 size sp ad res c bs 空间 stl
原文地址:http://www.cnblogs.com/huashiyiqike/p/3931674.html