Pytorch使用
torchtorch.tensor():用来生成已有的数据tensor(向量):a = torch.tensor([[2.3, 2, 5],[4, 5, 9.0]])print(a.shape)# print:torch.Size([2, 3])其他的tensor类型:注:与torch.tensor() 只接收现有数据,不能接收形状(shape)不同,下方可以接收形状也可以接收数据数据类型CPUGPU32 floattorch.FloatTensortorch
原创
2021-12-29 10:23:08 ·
541 阅读 ·
0 评论