你即將離開本站

並前往https://www.cxyzjd.com/article/u011622208/111634807

torch.argmax转到tensorrt之后输出结果出现1.5e-44_怡宝2号

描述将pytorch 的模型，模型的最后是torch.argmax 操作。转换成onxx，再用tensorrt 进行推理的时候， 结果不对，出现了1.5e-44 这样的数值。正确的应该都是整数才是。

確定！ 回上一頁

查詢 「Torchargmax」的人也找了：

  1. Pytorch maximum
    2.
  2. Tensor to int
    3.
  3. .item pytorch
    4.
  4. Torch topk
    5.
  5. Torch expand
    6.
  6. Torch .max outputs.data, 1
    7.
  7. Torch argsort
    8.
  8. Pytorch masked_fill
    9.