张量转换成Long类型
时间: 2023-11-16 20:07:08 浏览: 71
pytorch中tensor张量数据类型的转化方式
5星 · 资源好评率100%
在PyTorch中,可以使用`long()`函数将张量转换为长整型(int64)类型。示例代码如下:
```python
import torch
# 创建一个整型张量
x = torch.tensor([1, 2, 3, 4])
# 将整型张量转换为长整型张量
x = x.long()
# 打印结果
print(x)
```
输出结果为:
```
tensor([1, 2, 3, 4])
```
请注意,如果张量中的元素超出了长整型的范围,则会导致溢出错误。因此,在转换类型之前,请确保你的张量中的元素不会超出长整型的范围。
阅读全文