调用torch.Tensor类
时间: 2024-03-04 17:48:10 浏览: 63
pytorch实现Tensor变量之间的转换
5星 · 资源好评率100%
可以使用torch.Tensor类的构造函数创建张量。以下是一个创建张量的示例代码:
```
import torch
# 创建一个形状为2x3的空张量
my_tensor = torch.Tensor(2, 3)
print(my_tensor)
```
输出:
```
tensor([[1.1210e-44, 0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00]])
```
注意,这将创建一个随机初始化的张量。如果你需要其他类型的张量,可以使用torch中其他的构造函数,例如`torch.zeros()`,`torch.ones()`或`torch.randn()`等。此外,还可以使用`torch.tensor()`函数从Python列表或NumPy数组创建张量。
阅读全文