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])
相关推荐
MYH5167 分钟前
神经网络 隐藏层
人工智能·深度学习·神经网络
EverBule1 小时前
Python 训练 day46
开发语言·python
WangY_ZQ1 小时前
Python 如何在Python 3.6上安装PIP
linux·python·pip
聚客AI1 小时前
系统掌握PyTorch:图解张量、Autograd、DataLoader、nn.Module与实战模型
人工智能·pytorch·python·rnn·神经网络·机器学习·自然语言处理
狮子也疯狂1 小时前
基于Python的气象数据分析及可视化研究
python·信息可视化·数据分析
蓝婷儿2 小时前
6个月Python学习计划 Day 18 - 项目实战 · 学生成绩管理系统(OOP版)
开发语言·python·学习
HINOTOR_2 小时前
DAY 25 异常处理
开发语言·python
yorushika_2 小时前
python打卡训练营打卡记录day49
开发语言·python·tensorboard·cbam
qq_189370493 小时前
Jupyter notebook的文章结构目录查看方式和汉化方法
python·jupyter
晨曦5432103 小时前
将 Jupyter Notebook 的默认存储路径从 C 盘迁移到 D 盘,可以通过以下步骤实现:
ide·python·jupyter