torch.sum(image, axis=0)中axis含义及用法

ym62033

于 2024-04-24 10:55:52 发布

阅读量342

点赞数 1

CC 4.0 BY-SA版权

文章标签：人工智能

本文链接：https://blog.csdn.net/ym62033/article/details/138152384

默认情况下，调用求和函数会沿所有的轴降低张量的维度，使它变为一个标量。可以指定张量沿哪一个轴来通过求和降低维度。
以矩阵为例，通过求所有行元素的和来降维（轴0），即将相同列位置对应的所有行位置元素相加，列数保持不变，可以在调用函数时指定axis=0。

axis=1，是通过求所有列元素的和来降维(轴1)，即将相同行位置对应的所有列位置元素相加，行数保持不变。

import torch
T = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])
print("axis = 0:  ",torch.sum(T,axis = 0))
print("axis = 1:  ", torch.sum(T,axis = 1))

输出：
axis = 0:   tensor([12, 15, 18])
axis = 1:   tensor([ 6, 15, 24])