tensor补充

文章目录

矩阵*运算

python 复制代码
import torch
b = torch.arange(0,5000).unsqueeze(-1)
'''
tensor([[   0],
        [   1],
        [   2],
        ...,
        [4997],
        [4998],
        [4999]])
'''
a = torch.arange(0,128,2)
'''
tensor([  0,   2,   4,   6,   8,  10,  12,  14,  16,  18,  20,  22,  24,  26,
         28,  30,  32,  34,  36,  38,  40,  42,  44,  46,  48,  50,  52,  54,
         56,  58,  60,  62,  64,  66,  68,  70,  72,  74,  76,  78,  80,  82,
         84,  86,  88,  90,  92,  94,  96,  98, 100, 102, 104, 106, 108, 110,
        112, 114, 116, 118, 120, 122, 124, 126])
'''

c = b * a

'''
tensor([[     0,      0,      0,  ...,      0,      0,      0],
        [     0,      2,      4,  ...,    122,    124,    126],
        [     0,      4,      8,  ...,    244,    248,    252],
        ...,
        [     0,   9994,  19988,  ..., 609634, 619628, 629622],
        [     0,   9996,  19992,  ..., 609756, 619752, 629748],
        [     0,   9998,  19996,  ..., 609878, 619876, 629874]])
'''
相关推荐
头发够用的程序员17 分钟前
TensorRT 自定义算子插件实战(三):手搓 2×2 最大池化 customMaxpool
人工智能·pytorch·python·深度学习·神经网络·边缘计算·jetson
俊哥V28 分钟前
AI一周事件 · 2026-09-30 至 2026-10-06
人工智能·ai
wflynn11 小时前
Reflection AI 发布 501B 参数开放权重模型 Beam
ai·技术
xiangshangdemayi11 小时前
Strata教程:让GISer真正实现了token自由
ai·gis·教程·token·strata·自由
空堂与归12 小时前
Hinton 首篇 RSI 论文:AI 自我进化怎么闭环?
人工智能·ai
软件开发技术深度爱好者12 小时前
DeepSeek Harnesss使用教程
ai·技术实践
2601_9567436812 小时前
上海GEO营销工程实践:知识库构建、AI内容生产、官网承接与监测优化闭环评估
人工智能·ai·geo·上海
空心木偶☜14 小时前
LangGraph 错误处理和重试机制
ai·ai编程·langgraph
菩提小狗15 小时前
每日极客日报 · 2026年10月06日
ai·开源·极客日报·it热点·技术资讯
林伽一17 小时前
能力趋同、账单分化,技术选型正在从榜单转向负载|2026年10月06日
人工智能·科技·安全·ai