在进行离线推理的时候,run之后获得的数据类型为tensor,而不是numpy,请问如何拿到tensor?
收藏回复举报
在进行离线推理的时候,run之后获得的数据类型为tensor,而不是numpy,请问如何拿到tensor?
t('forum.solved') 已解决
发表于2024-03-26 08:39:42
0 查看

在用ais_bench.infer.interface.InferSession进行离线推理,输入到run中的数据都是numpy()类型,运行完run之后,获取到的数据类型却为tensor,打印出来显示如下:

<Tensor> 

shape:  (1, 256, 768) 

dtype:  float32 

device: 0

用torch.save保存,又会出现报错:TypeError: cannot pickle 'aclruntime.Tensor' object。应该如何获取tensor里面的数据呢?

我要发帖子