微凉的衣柜 2021-06-03 00:16 采纳率: 63.9%
浏览 295
已采纳

(Libtorch) FP16 tensor 错误 C10error

I got c10 error when I define a FP16 tensor and use torch::max() and torch::sqrt(). The code are shown as follows:

当我定义一个FP16的tensor并使用torch::max() 和 torch::sqrt()函数时会出现C10 error错误,代码如下:

	float test_float[3][3] = { {1.0, 2.0, 3.0}, {4.0, 5.0, 6.0 },{7.0, 8.0, 9.0} };
	torch::Tensor test_float_tensor = torch::from_blob(test_float, { 3, 3 }).to(at::kCPU).to(torch::kFloat16);
	torch::sqrt(test_float_tensor);
	torch::max(test_float_tensor, 1, true);

错误如下:

当把tensor改为FP32或FP64,错误就消失了,代码如下:

	float test_float[3][3] = { {1.0, 2.0, 3.0}, {4.0, 5.0, 6.0 },{7.0, 8.0, 9.0} };
	torch::Tensor test_float_tensor = torch::from_blob(test_float, { 3, 3 }).to(at::kCPU).to(torch::kFloat32);
	torch::sqrt(test_float_tensor);
	torch::max(test_float_tensor, 1, true);

为什么会造成这种错误呢?采用FP16 tensor,执行torch::sum()函数时并不会出现这个错误。是FP16 tensor有什么不适用的地方吗?求解答,谢谢!

  • 写回答

1条回答 默认 最新

报告相同问题?

悬赏问题

  • ¥15 MATLAB代码补全插值
  • ¥15 Typegoose 中如何使用 arrayFilters 筛选并更新深度嵌套的子文档数组信息
  • ¥15 前后端分离的学习疑问?
  • ¥15 stata实证代码答疑
  • ¥50 husky+jaco2实现在gazebo与rviz中联合仿真
  • ¥15 dpabi预处理报错:Error using y_ExtractROISignal (line 251)
  • ¥15 在虚拟机中配置flume,无法将slave1节点的文件采集到master节点中
  • ¥15 husky+kinova jaco2 仿真
  • ¥15 zigbee终端设备入网失败
  • ¥15 金融监管系统怎么对7+4机构进行监管的