PyTorch基本使用方法

发布时间:2024年01月24日

创建一个矩阵:

x = torch.empty(5, 3)
x

创建一个全0矩阵:

x = torch.zeros(3, 3, dtype=torch.long)
x

传入一个数据:

x = torch.tensor([5.5, 3])
x

显示矩阵大小:

基本计算方法:

torch.add(x,y)
x+y
#都是加法操作

索引:

x[:, 1]

冒号表示取所有,数字表示取哪一个

view操作改变矩阵维度:

x.view(16),如果原来矩阵是4*4的那么现在就是1*16的

x.view(-1,8),-1表示自动计算,计算每列8行会有几列,.size()后会显示[2,8]

与numpy协同操作,可以把tensor格式通过.numpy()转换格式,同时numpy格式也可以通过torch.from_numpy(n)转换格式

文章来源:https://blog.csdn.net/weixin_55843921/article/details/135702510
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。