做个笔记,在用 libtorch 二次开发的时候,需要查看 tensor 的值是否正确,打印方式如下:
// print tensor 打印前20 个
torch::Tensor grad_bottom_tensor
void *ptr = grad_bottom_tensor.data_ptr();
for (size_t i = 0; i < 20; i++) {
std::cout<<*((float*)(ptr+i))<<std::endl;
}
做个笔记,在用 libtorch 二次开发的时候,需要查看 tensor 的值是否正确,打印方式如下:
// print tensor 打印前20 个
torch::Tensor grad_bottom_tensor
void *ptr = grad_bottom_tensor.data_ptr();
for (size_t i = 0; i < 20; i++) {
std::cout<<*((float*)(ptr+i))<<std::endl;
}