pytroch基础用法

torch.randn（size)

torch.randn:用来生成随机数字的tensor，这些随机数字满足标准正态分布（0~1）

import torch
a=torch.randn(3)
b=torch.randn(3,4)
print("a:",a)
print("b:",b)

a: tensor([ 0.9405, -0.1068,  0.1712])
b: tensor([[-1.0962, -0.1893,  1.2323,  0.5703],
        [-1.5256, -1.4923,  0.4275,  0.5143],
        [ 1.1200,  0.5317,  1.1961, -2.2533]])

torch.full(size, fill_value)

import torch
a = torch.full([2,3],2.0)
print("a:", a)

a: tensor([[2., 2., 2.],
        [2., 2., 2.]])

troch.arange() / torch.range()

torch.linspace()/torch.logspace()

torch.ones()/torch.zeros()/torch.eye()

torch.eye(): 输出3 x 3的矩阵 对角线全为1

torch.randperm()

将0~n-1（包括0和n-1）随机打乱后获得的数字序列

维度变换

View reshape

torch.unsqueeze()

torch.unsqueeze(input, dim, out=None)

作用：扩展维度
返回一个新的张量，对输入的既定位置插入维度 1

torch.squeeze()

对数据的维度进行压缩
torch.squeeze(input, dim=None, out=None)

torch.expand()

  expand（）函数的功能是用来扩展张量中某维数据的尺寸，它返回输入张量在某维扩展为更大尺寸后的张量。
  扩展张量不会分配新的内存，只是在存在的张量上创建一个新的视图（关于张量的视图可以参考博文：由浅入深地分析张量），而且原始tensor和处理后的tensor是不共享内存的。
  expand()函数括号中的输入参数为指定经过维度尺寸扩展后的张量的size。

(1)# 在行上更改
import torch
a = torch.tensor([1, 2, 3])   # C：一行三列
c = a.expand(2, 3) # 将C进行扩为：两行三列
print(a)
print(c)
# 输出信息：
tensor([1, 2, 3])
tensor([[1, 2, 3],
        [1, 2, 3]]
(2) # 在列上更改
import torch
a = torch.tensor([[1], [2], [3]])  # C：三行一列
print(a.size())
c = a.expand(3, 3)  # 将C进行扩为：三行三列
print(a)
print(c)
# 输出信息：
torch.Size([3, 1])
tensor([[1],
        [2],
        [3]])
tensor([[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]])

torch.repeat()

PyTorch中的repeat()函数可以对张量进行复制。
当参数只有两个时，第一个参数表示的是复制后的列数，第二个参数表示复制后的行数。
当参数有三个时，第一个参数表示的是复制后的通道数，第二个参数表示的是复制后的列数，第三个参数表示复制后的行数。

>>> x = torch.tensor([6,7,8])
>>> x.repeat(4,2)
tensor([[6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8],
        [6, 7, 8, 6, 7, 8]])
>>> x.repeat(4,2,1)
tensor([[[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]],

        [[6, 7, 8],
         [6, 7, 8]]])
>>> x.repeat(4,2,1).size()
torch.Size([4, 2, 3])

torch.permute()

维度变化

torch.transpose()

transpose可以理解为矩阵转置概念在高维数组上的扩展

from torch import Tensor
import numpy as np
x = np.random.rand(3, 2, 5, 6)
z = Tensor(x)
# z.shape is (3, 2, 5, 6)
print(z.transpose(2, 3).shape)

torch.Size([3, 2, 6, 5])

torch.t()

torch.t()是一个类似于求矩阵的转置的函数，但是它要求输入的tensor结构维度<=2D

>>> x = torch.randn(()) ##0D
>>> x
tensor(0.1995)
>>> torch.t(x)
tensor(0.1995)
>>> x = torch.randn(3) ## 1D
>>> x
tensor([ 2.4320, -0.4608,  0.7702])
>>> torch.t(x)  ### torch version =1.0.1 报错
tensor([.2.4320,.-0.4608,..0.7702])
>>> x = torch.randn(2, 3)  ##2D size(2,3)
>>> x
tensor([[ 0.4875,  0.9158, -0.5872],
        [ 0.3938, -0.6929,  0.6932]])
>>> torch.t(x)  ## size(3,2)
tensor([[ 0.4875,  0.3938],
        [ 0.9158, -0.6929],
        [-0.5872,  0.6932]])

拼接与拆分

torch.cat()


torch.stack()

torch.cat()

dim控制拼接的维度，dim=0为上下拼接，dim=1为左右拼接[1

>>> import torch
>>> a = torch.rand((2,3))
>>> a
tensor([[0.7515, 0.1021, 0.0726],
        [0.0575, 0.1666, 0.2763]])
>>> b = torch.rand((2,3))
>>> b
tensor([[0.7485, 0.8340, 0.2617],
        [0.7847, 0.2847, 0.3445]])
>>> c =torch.cat((a,b),dim=0)
>>> c
tensor([[0.7515, 0.1021, 0.0726],
        [0.0575, 0.1666, 0.2763],
        [0.7485, 0.8340, 0.2617],
        [0.7847, 0.2847, 0.3445]])
>>> d = torch.cat((a,b),dim=1)
>>> d
tensor([[0.7515, 0.1021, 0.0726, 0.7485, 0.8340, 0.2617],
        [0.0575, 0.1666, 0.2763, 0.7847, 0.2847, 0.3445]])

torch.split()

import torch
x = torch.randn(1, 2, 4, 4)
y = torch.split(x, 1, dim=2)  # 每块大小为1
# print(x[0])
for i in y:
    print(i.size())

a = torch.rand(1, 4, 8, 6)
b = torch.split(a, 2, dim=1)  # 每块大小为2
for i in b:
    print(i.size())

torch.Size([1, 2, 1, 4])
torch.Size([1, 2, 1, 4])
torch.Size([1, 2, 1, 4])
torch.Size([1, 2, 1, 4])
torch.Size([1, 2, 8, 6])
torch.Size([1, 2, 8, 6])

torch.chunk()

import torch
a=torch.tensor([[[1,2],[3,4]],
               [[5,6],[7,8]]])
print(a.size())
b=torch.chunk(a,2,1)
print("a:", a)
print("b:", b)

torch.Size([2, 2, 2])
a: tensor([[[1, 2],
         [3, 4]],
        [[5, 6],
         [7, 8]]])
b: (tensor([[[1, 2]],
        [[5, 6]]]), tensor([[[3, 4]],
        [[7, 8]]]))

torch.add()/ +

torch.matmul()/troch.mm()/@