c++ libtorch 多维张量和arrayfire 多维数组

torch::Tensor r = torch::rand({2,3,4,5});
std::cout << r.options().dtype();

//torch::rand({2,3,row,col})

第3维度是矩阵的行，第4维度是矩阵的列，以上代码表示生成2*3个4行5列的默认为torch::float32数据类型的随机矩阵

af::array a = af::randu(2 , 3, 4, 5);
//af::randu(row,col, 4, 5)

第1维度是矩阵的行，第2维度是矩阵的列，以上代码表示生成4*5个2行3列的默认为f32数据类型的随机矩阵