Pytorch简单一览表

import torch
import numpy as np

#3. 创建一个长度为 10，每个元素值都为 0 的一维 Tensor (★☆☆)
torch.zeros([10])

#6. 创建一个长度为 10 的一维 Tensor，除了第五个元素是 1 外，其他元素都是 0(★☆☆)
# 输出如右：tensor([0., 0., 0., 0., 1., 0., 0., 0., 0., 0.])
t = torch.zeros([10])
t[4] = 1
t

#7. 创建一个元素值依次为 10 到 49 的一维 Tensor(★☆☆)
#tensor([10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27,
        # 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45,
        # 46, 47, 48, 49])
torch.arange(10, 50)

#9. 创建一个 3*3 的二维 Tensor，值为 0 到 8 (★☆☆)
# tensor([[0, 1, 2],
#         [3, 4, 5],
#         [6, 7, 8]])

#10. 获取 [1,2,0,0,4,0] 中非零元素的下标(★☆☆)
# tensor([[0],
#         [1],
#         [4]])
t = torch.Tensor([1,2,0,0,4,0])
t.nonzero()

#11. 创建一个 3*3 的 单位矩阵（identity matrix） (★☆☆)
# tensor([[1., 0., 0.],
#         [0., 1., 0.],
#         [0., 0., 1.]])
torch.eye(3)

#12. 创建一个 3x3x3 的随机 Tensor (★☆☆)
# tensor([[[0.3535, 0.6218, 0.2071],
#          [0.4987, 0.3297, 0.3077],
#          [0.8799, 0.7423, 0.5187]],
#
#         [[0.1057, 0.6544, 0.6071],
#          [0.3751, 0.9311, 0.5071],
#          [0.7958, 0.1009, 0.9900]],
#
#         [[0.5937, 0.0500, 0.8445],
#          [0.2194, 0.5079, 0.4057],
#          [0.4983, 0.8374, 0.4039]]])
torch.rand([3,3,3])

#13. 创建一个 10x10 的随机 Tensor，并找出其中的最大值和最小值(★☆☆)
t = torch.rand([10, 10])
print(t.max())
print(t.min())

#14. 创建一个长度为 30 的随机一维 Te