第1关:Tensor数据类型

import torch

x = torch.rand(5, 3)

########## Begin ##########

tensor_size = torch.Size([5, 3])
########## End ##########


print(tensor_size)

第2关:创建方法

import torch

########## Begin ##########

a_tensor = torch.tensor([5, 3, 2]) 
########## End ##########

print(a_tensor) 

第3关:Session

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐