PyTorch_创建01张量

• torch.ones 和 torch.ones_like          创建全1张量
  
• torch.zeros 和 torch.zeros_like        创建全0张量
  
• torch.full 和 torch.full_like          创建全为指定值张量
  

上面的函数里有 like 表示根据别的一个张量的外形创建全0大概全1的大概全为指定值的张量。

---

代码

1. import torch
2. import numpy as np
4. # 创建全为0的张量
5. def test01():
6.     # 创建指定形状全为0的张量
7.     data = torch.zeros(2, 3)
8.     print(data)
10.     # 根据其他张量的形状去创建全0张量
11.     data = torch.zeros_like(data)
12.     print(data)
14. # 创建全为1的张量
15. def test02():
16.     # 创建指定形状全为1的张量
17.     data = torch.ones(2, 3)
18.     print(data)
20.     # 根据其他张量的形状去创建全1张量
21.     data = torch.ones_like(data)
22.     print(data)
24. # 创建全为指定值的张量
25. def test03():
26.     # 创建形状为2行3列，值全部为10的张量
27.     data = torch.full([2, 3], 10)
28.     print(data)
30.     # 创建形状和data一样，但是值全部为20的张量
31.     data = torch.full_like(data, 20)
32.     print(data)
34. if __name__ == '__main__':
35.     test02()

复制代码