亲测解决返回的batch是一个list而不是长度为1的数值

admin2024-09-05 26

这个问题是由于tensor定义错误导致的，解决方法是用torch.tensor

问题原文

[1.0900066e-08 0.0000000e+00 1.1666273e-08 0.0000000e+00]

将

Input = Tensor(Input)

改成


Input = torch.tensor(Input)

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明原文出处。如若内容造成侵权/违法违规/事实不符，请联系SD编程学习网：675289112@qq.com进行投诉反馈，一经查实，立即删除！