pytorc创建一个矩阵

创建一个Tensor（很常用）

import torch

创建一个全1矩阵

x = torch.ones(5,5)
>>> x
tensor([[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]])
>>> 

创建一个随机矩阵

x = torch.rand(2,3) 
>>> x
tensor([[0.9798, 0.6616, 0.8331],
        [0.0387, 0.4479, 0.9542]])
>>> 

初始化一个全零的矩阵

x = torch.zeros(5,3,dtype = torch.int)  # 32位整型
>>> x
tensor([[[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]],

        [[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]],

        [[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]],

        [[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]],

        [[0, 0, 0],
         [0, 0, 0],
         [0, 0, 0]]], dtype=torch.int32)
>>> 

查看矩阵尺寸

x.size()
>>> x.size()
torch.Size([3, 3, 3, 3])
>>>