python乱炖6——sum(),指定维度进行求和

python乱炖6------sum(),指定维度进行求和

py 复制代码
import torch

# 创建一个三维张量
x = torch.tensor([
    [[1, 2, 3], [4, 5, 6]],
    [[7, 8, 9], [10, 11, 12]]
])

print("Original tensor x:")
print(x)
print(x.shape)

>>> tensor([[[ 1,  2,  3],
         [ 4,  5,  6]],

        [[ 7,  8,  9],
         [10, 11, 12]]])
    torch.Size([2, 2, 3])



# 沿第0维求和
sum_dim0 = torch.sum(x, dim=0)
print("Sum along dimension 0:")
print(sum_dim0)
print()

>>>Sum along dimension 0:
tensor([[ 8, 10, 12],
        [14, 16, 18]])




# 沿第1维求和
sum_dim1 = torch.sum(x, dim=1)
print("Sum along dimension 1:")
print(sum_dim1)
print()

>>>Sum along dimension 1:
tensor([[ 5,  7,  9],
        [17, 19, 21]])




# 沿第2维求和
sum_dim2 = torch.sum(x, dim=2)
print("Sum along dimension 2:")
print(sum_dim2)
print()

>>>Sum along dimension 2:
tensor([[ 6, 15],
        [24, 33]])





sum_dim3 = torch.sum(x, dim=(1,2))
print("Sum along dimension (1,2):")
print(sum_dim3)
print()

>>>Sum along dimension (1,2):
   tensor([21, 57])





sum_dim4 = torch.sum(x, dim=(0,1))
print("Sum along dimension (0,1):")
print(sum_dim4)
print()
>>>Sum along dimension (0,1):
tensor([22, 26, 30])
相关推荐
飞翔的佩奇3 小时前
【完整源码+数据集+部署教程】【天线&水】舰船战舰检测与分类图像分割系统源码&数据集全套:改进yolo11-repvit
前端·python·yolo·计算机视觉·数据集·yolo11·舰船战舰检测与分类图像分割系统
木头左4 小时前
最大回撤约束下ETF多因子动态止盈参数校准方案
python
汤姆yu4 小时前
2026版基于python的协同过滤音乐推荐系统
开发语言·python
汤姆yu4 小时前
基于python的电子商务管理系统
开发语言·python
may_一一4 小时前
pycharm\python 安装下载
ide·python·pycharm
后台开发者Ethan5 小时前
FastAPI之 Python的类型提示
python·fastapi·ai编程
hello kitty w5 小时前
Python学习(11) ----- Python的泛型
windows·python·学习
没有梦想的咸鱼185-1037-16635 小时前
AI Agent结合机器学习与深度学习在全球气候变化驱动因素预测中的应用
人工智能·python·深度学习·机器学习·chatgpt·数据分析
周杰伦_Jay5 小时前
【MCP开发部署流程表格分析】MCP架构解析、开发流程、部署方案、安全性分析
人工智能·深度学习·opencv·机器学习·架构·transformer
晚霞apple6 小时前
多模态大模型的前沿算法综述
论文阅读·人工智能·深度学习·神经网络·机器学习