目录
c++ 打印 torch tensor
for (auto& tensor : m_OutputTensors)
{
for (int a = 0; a < 20; a++) {
std::cout << tensor.hostBuffer[a] << " ";
}
std::cout << std::endl;
}
新版是at::Tensor
auto top_scores_a = image_prediction_data[m][6].item().toInt();
printf("111111111111111:%d \n",top_scores_a);
cuda cu代码中打印:
printf("%d\n",mask[ph * poo