Pytorch学习教程:Tensor

Tensor对于pytorch就如同ndarray,而且和ndarray很像,但有一个重要的区别就是Tensor支持GPU,ndarray不支持。


注意为了说明方便,
1.Tensor分类:
ByteTensor,唯一的一个无符号类型Tensor,8位无符号整型(下面皆有符号)
CharTensor,8位有符号整型
ShortTensor,16位整型
IntTensor,32位整型
LongTensor,64位整型
FloatTensor,32位浮点,默认类型
DoubleTensor,64位浮点


2.初始化
(1)通过其他类型对象初始化
torch.Tensor([[1,2],[3,4]])
torch.from_numpy(ndarray) 返回的Tensor与参数ndarray共享内存

(2) 通过函数初始化
torch.zeros((n1,n2…),out=None)用元组传入维数信息
torch.eye(n,m=None,out=None) 只能返回2维对角阵
很多函数运算结果也会返回一个Tensor
(3)随机初始化

3.获取Tensor的相关参数

torch.numel(a)获取元素个数
a.size()返回一个torch.Size类型,可以当作tensor,支持索引

4.类型转换
a.float()等即可
a.numpy()可转为ndarray,同样也共享内存

5.获取及修改
支持索引和切片。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值