本文整理汇总了C++中tensor::shape方法的典型用法代码示例。如果您正苦于以下问题:C++ tensor::shape方法的具体用法?C++ tensor::shape怎么用?C++ tensor::shape使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensor
的用法示例。
在下文中一共展示了tensor::shape方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: print_2D_view
void print_2D_view(tensor<T, M>& a) {
for (unsigned int i = 0; i<a.shape(0); i++){
for (unsigned int j = 0; j<a.shape(1); j++)
cout<<a(i, j)<<" ";
cout<<endl;
}
}
示例2: print_3D_view
void print_3D_view(tensor<T, M>& a) {
for (unsigned int k = 0; k<a.shape(0); k++) {
cout<<"dimension 0: "<<k<<endl;
for (unsigned int i = 0; i<a.shape(1); i++){
for (unsigned int j = 0; j<a.shape(2); j++)
cout<<a(k, i, j)<<" ";
cout<<endl;
}
}
}
示例3: rand_init
void rand_init(tensor<T, M>& a){
for (unsigned int i=0; i<a.shape(0);i++)
a[i] = drand48();
}