如何从张量中获取一个值?

问题描述 投票:0回答:1

我的python函数返回的是 tensor([102],cuda::0) 样的结构。我如何从中获取第一个值?

pytorch tensor
1个回答
0
投票

发现这个 pytorch论坛:

a.data.cpu().numpy()[0]

其中 a 是张量。

© www.soinside.com 2019 - 2024. All rights reserved.