torch.Tensor()创建张量
一定要记得
a = torch.Tensor(1,2,3) 表示创建一个尺寸为1,2,3的张量,取值为随机
如果要创建特定元素,要使用torch.tensor(1),注意是tensor全小写
比如,a = torch.Tensor([1,2,3]),打印后表示tensor([1,2,3])
********************************************************************
在 PyTorch 中,torch.Tensor()
和 torch.tensor()
都可以用来创建张量(Tensor),但它们有一些区别,包括用法和行为:
-
torch.Tensor():
torch.Tensor()
是一个构造函数,可以用于创建一个未初始化的张量。- 它会创建一个具有默认数据类型(通常是
float32
)的张量。 - 不建议使用
torch.Tensor()
来创建张量,因为它可能会导致不可预测的结果,而且在某些版本的 PyTorch 中,可能会改变其行为。
-
torch.tensor():
torch.tensor()
是一个函数,可以根据提供的数据创建一个新的张量。- 它会根据提供的数据自动推断张量的数据类型,并可以通过
dtype
参数进行显式设置。 - 通常,使用
torch.tensor()
来创建张量是更好的选择,因为它提供了更明确和一致的行为。
以下是使用两者的示例代码:
使用 torch.Tensor()
创建张量(不建议):
import torch
# 不建议使用这种方式创建张量
tensor = torch.Tensor([1, 2, 3])
print("张量:", tensor)
使用 torch.tensor()
创建张量:
import torch
# 推荐使用这种方式创建张量
tensor = torch.tensor([1, 2, 3])
print("张量:", tensor)
总之,为了避免不必要的混淆并确保代码的可读性,建议使用 torch.tensor()
来创建张量。
本文来自博客园,作者:海_纳百川,转载请注明原文链接:https://www.cnblogs.com/chentiao/p/17657579.html,如有侵权联系删除
分类:
pytorch
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
2022-08-25 linux中查看包含某个关键字的文件夹中的文件数量
2022-08-25 如何在 xargs 命令中使用 >