如果你想从 Tensor 中提取单个元素,可以使用 Tensor 的 item()
方法。这个方法会将 Tensor 转换成 Python 的标量类型(例如 int、float 等)并返回其值。
例如,如果你有一个 Tensor t
,你可以这样提取它的第一个元素:
x = t[0]
y = x.item()
如果你想从 Tensor 中提取单个元素,可以使用 Tensor 的 item()
方法。这个方法会将 Tensor 转换成 Python 的标量类型(例如 int、float 等)并返回其值。
例如,如果你有一个 Tensor t
,你可以这样提取它的第一个元素:
x = t[0]
y = x.item()