PyTorch初学
初识PyTorch 1.张量 import torch #创建一个空的5x3张量 x = torch . empty ( 5 , 3 ) #创建一个随机初始化的5x3张量 x = torch . rand ( 5 , 3 ) # 创建一个5x3的0张量,类型为long x = torch . zeros ( 5 , 3 , dtype = torch . long ) # 创建一个5x3的单位张量,类型为double x = torch . ones ( 5 , 3 , dtype = torch . double ) # 直接从数组创建张量 x = torch . tensor ( [ 5.5 , 3 ] ) #从已有的张量创建相同维度的新张量,并且重新定义类型为float x = torch . randn_like ( x , dtype = torch . float ) # 打印一个张量的维度 print ( x . size ( ) ) # 将两个张量相加 y = torch . rand ( 5 , 3 ) print ( x + y ) print ( torch . add ( x , y ) ) result = torch . zeros ( 5 , 3 ) torch . add ( x , y , out = result ) print ( result