实践掌握深度学习 | 核心Python函数解析

阅读量：

(1):维度dim，保持原有维度keepdim

以下通过图像形式直观呈现了这两个参数的具体功能与影响。

复制代码

    X = torch.tensor([[1, 2, 3], [4, 5, 6]])
    print(X.sum(dim=0, keepdim=True))
    print(X.sum(dim=1, keepdim=True))
    print(X.sum(dim=0, keepdim=False))
    print(X.sum(dim=1, keepdim=False))

(2):view函数的用法

view函数的使用方式如下，其主要功能是调整tensor的维度结构。在该函数中，参数-1的作用是让当前维度的大小根据其他已指定的维度进行自

全部评论 (0)

还没有任何评论哟~