提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  后端

当前栏目

【PyTorch】取出tensor中的值

PyTorch 取出 Tensor
2023-09-14 09:12:47 时间

【PyTorch】取出tensor中的值

1、背景

在使用pytorch进行运算时，pytorch中数据经常以tensor形式出现。
如果你还不熟悉tensor的操作，可以将tensor的值取出来做分析。
操作关键字：
tensor.item()

2、示例

import torch
print(torch.__version__)
a = torch.rand(3, 2)
print

相关文章