har*_*rsh 5 python torch tensorflow pytorch tensor
我有一个张量列表,我想将其转换为浮点,我该怎么做。我尝试过使用.item但它不起作用。我正进入(状态ValueError: only one element tensors can be converted to Python scalars。
tensor([[12.1834, 4.9616, 7.7913],
[ 8.9394, 8.5784, 9.3691],
[ 9.4475, 8.9766, 9.8418],
[11.8210, 6.0852, 8.2168],
[ 8.2885, 6.2607, 9.8877]], grad_fn=<CloneBackward0>)
Run Code Online (Sandbox Code Playgroud)
小智 9
您只需将 Tensor 常量转换为 numpy 对象,然后就可以通过索引访问。
result.numpy()[0]
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
37591 次 |
| 最近记录: |