Pytorch训练问题:torch.tensor()函数的解释及相关思考

这篇博客是我对于torch框架中的tensor的一些记录和思考,会不定时更新我新的想法。

1. 介绍一下torch.tensor()这个函数

在 PyTorch 中,torch.tensor 是一个多维矩阵,其中包含同种类型的元素(如整数或浮点数)。它是 PyTorch 库的核心类,几乎所有的计算操作都可以在 tensor 上进行。

注意在torch中小写的tensor和大写的Tensor是不同的用法,可见下列是创建方式:

import torch

#  这是小写的tensor
x = torch.tensor([1.0, 2.0, 3.0])

#  这是大写的Tensor
y = torch.Tensor(2,3)

在创建中,tensor是直接可以输入数据创建,Tensor是创建---有几个维度,每个维度多大的函数。

2. 对于tensor函数的应用

一般使用在矩阵中,那么矩阵的行列怎么在函数中体现,下面我只讨论在二维的tensor中,只有二维才会存在行列的概念。

我一般在二维的时候将数据编写方式进行改变来更好的看数据。

#  这个一个二维tensor的输入
output = torch.tensor([[1.0, 2.0, 3.0],[2.0, 3.0, 4.0]])

#  输出后可以获得维度 [2,3]
print(output.shape)

#  修改后的数据形式
#  行和列可以清晰的通过观察感知到,对于去理解对应的矩阵变换公式有好处
output = torch.tensor([
    [1.0, 2.0, 3.0],
    [2.0, 3.0, 4.0]
])

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值