tensor数据--边做边学笔记

一、相关函数

 torch.zeros(a，b, dtype = )

函数含义：创建有a行b列的多维数据，数据全是0

数据形式：

tensor（[0,0,0,0,....,0]

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

                        ...

               [0,0,0,0,....,0]）

二、基本操作

1. tensor数据之间可以实现批量赋值与运算

元组、列表只可以实现单个元素赋值，数组与tensor数据都可以实现批量赋值与运算，也就是按行或列进行赋值或者运算

 按列赋值（一列一列地赋值）

import torch
ob = torch.zeros(2,3, dtype=int)
print(ob)
a = [(1, 2), (3, 4)]
a = torch.tensor(a, dtype=int)
print(a)
ob[:, (0,1)] = a
print(ob)


控制台：
tensor([[0, 0, 0],
        [0, 0, 0]])
tensor([[1, 2],
        [3, 4]])
tensor([[1, 2, 0],
        [3, 4, 0]])

按列运算

import torch
ob = torch.zeros(2,3, dtype=int)
a = [(1, 2), (3, 4)]
a = torch.tensor(a, dtype=int)
print(ob)
print(a)
ob[:, (0,1)] = a
print(ob)
c = ob[:, 0]**2 + ob[:, 1]**2
print(c)
ob[:, 0] += 1
print(ob)

控制台：

tensor([[0, 0, 0],
        [0, 0, 0]])
tensor([[1, 2],
        [3, 4]])
tensor([[1, 2, 0],
        [3, 4, 0]])
tensor([ 5, 25])
tensor([[2, 2, 0],
        [4, 4, 0]])