使用torch中的方法创建二维tensor
时间: 2024-09-09 15:05:26 浏览: 46
pytorch中tensor的合并与截取方法
在PyTorch中,创建二维tensor可以使用多种方法,其中一种常见的方法是使用`torch.tensor()`函数,并通过指定数据类型来创建二维tensor。下面是一个创建二维tensor的例子:
```python
import torch
# 创建一个二维tensor,这里以创建一个3x4的二维tensor为例
data = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]
tensor_2d = torch.tensor(data)
print(tensor_2d)
```
执行上述代码后,会得到一个3行4列的二维tensor。此外,还可以使用其他方法创建二维tensor,例如使用`torch.zeros()`、`torch.ones()`、`torch.randint()`等函数来创建特定值的二维tensor。
例如,创建一个3x5的全零二维tensor:
```python
tensor_2d_zeros = torch.zeros(3, 5)
```
或者创建一个5x3的全一二维tensor:
```python
tensor_2d_ones = torch.ones(5, 3)
```
阅读全文