水铝英石

吃书-2章-pytorch基础

1-Tensor

数据类型CPU TensorGPU Tensor
32位floattorch.FloatTensortorch.cuda.FloatTensor
64位floattorch.DoubleTensor
16位半精度floatN/A
8位半精度floattorch.ByteTensor
8位有符号inttorch.CharTensor
16位有符号inttorch.ShortTensor
32位有符号inttorch.IntTensor
64位有符号inttorch.LongTensor
torch.set_default_tensor_type(‘楼上的tensor类型’) 如torch.FloatTensor
import torch

a=torch.Tensor(2, 2)  #创建tensor对象,默认是FloatTensor
a 
b = a.double()
b
c = a.type(torch.DoubleTensor)
c
d = a.type_as(b)
d

发表评论

邮箱地址不会被公开。 必填项已用*标注

滚动到顶部