torch.tensor 与 torch.Tensor 的区别
创始人
2025-05-28 18:05:15

Tensor 和tensor 都能用于生成新的张量

torch.Tensor()

         python类,默认张量类型torch.FloatTensor()的别名, torch.Tensor([1,2])会调用Tensor类的构造函数__init__,生成单精度浮点类型的张量。

a = torch.Tensor([1, 2])
print(a)  # tensor([1., 2.])
print(type(a))  # 
print(a.type())  # torch.FloatTensor

torch.tensor()

         python函数,torch.tensor(data, dtype=None, device=None, requires_grad=False) data可以是:list, tuple, NumPy ndarray, scalar和其他类型。 torch.tensor会从data中的数据部分做拷贝(而不是直接引用), 根据原始数据类型生成相应的torch.LongTensor、torch.FloatTensor和torch.DoubleTensor。

b = torch.tensor([1, 2])
print(b)  # tensor([1, 2])
print(type(b))  # 
print(b.type())  # torch.LongTensorcc = np.zeros(2, dtype=np.float64)
c = torch.tensor(cc)
print(c.type())  # torch.DoubleTensor

torch.empty()

         生成指定类型、指定设备以及其他参数的张量, torch.Tensor()只能指定数据类型为torch.float, torch.Tensor()可以看做torch.empty()的一个特殊情况。

e = torch.empty(size=(3, ))
print(e)  # tensor([0., 2., 0.])
e2 = torch.empty(3)
print(e2)  # tensor([9.2196e-41, 0.0000e+00, 0.0000e+00])

特殊情况

s1 = torch.tensor(1)
print(s1)  # tensor(1)
print(s1.type())  # torch.LongTensor
s2 = torch.Tensor(1)
print(s2)  # tensor([0.])
print(s2.type())  # torch.FloatTensor

参考:

【PyTorch】Tensor和tensor的区别_玄云飘风的博客-CSDN博客

torch — PyTorch 1.13 documentation

相关内容

热门资讯

外媒:新加坡两辆双层巴士相撞致... 【环球网报道 记者 李梓瑜】据新加坡《联合早报》《海峡时报》报道,新加坡裕廊西地区当地时间14日上午...
“永远50年”魔咒终结   炒股就看金麒麟分析师研报,权威,专业,及时,全面,助您挖掘潜力主题机会! (来源:虎嗅APP)...
妙可蓝多董事张平退休离任 沈... 12月14日,妙可蓝多(600882)发布公告,董事张平因届满退休年龄提前离任,原定离任时间为202...
福州高新区主要领导走访服务园区... 12日上午,福州高新区党工委书记李贵勇、管委会主任林勉建走访服务园区重点企业和创新平台,宣讲党的二十...
开封万岁山武侠城演员被冻成“冰... 近日,河南多地下雪,一游客在开封万岁山武侠城游玩时,发现雪地里一座关羽的“冰雕”,走近才发现是一位正...