torch.tensor和torch.Tensor

在torch的tensor使用过程中,我们经常回看到torch.tensor()和类似于torch.Tensor()或者torch.FloatTesnor()、tensor.DoubleTensor()等等,究竟两种用法该怎么区分呢?

首先,要说的是,上面列出的几种初始化方法,我们统一的归为两类:大写和小写

小写:torch.tensor()

大写:torch.Tensor()或者torch.FloatTesnor()、tensor.DoubleTensor()诸如此类用Tensor开头的

小写一般只接受具体的数据作为输入,初始化方法举例来说:

1、用来定义标量时,不加[]即可:

print(torch.tensor(1.300))

print(torch.tensor(1.))

 

2、用来定义向量时,加[]

print(torch.tensor([1.1, 2.2]))

 

大写:既可以接受具体数据作为输入,也可以是维度。什么意思?请看例子:

1、使用具体数据作为输入:

torch.FloatTensor([1.1, 2.2])

 

2、维度作为输入,生成随机数据

torch.FloatTensor(1,2,3)#1,2,3是shape

 

为了避免混淆,建议:

tensor是接受具体的数据作为输入,而Tensor只用来接受维度作为输入,以避免混淆

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值