Skip to content

tensorlayerx.argmax没有参数keepdim,但paddle.argmax有 #45

@zhxiucui

Description

@zhxiucui

paddle源码paddle.argmax(x, axis=None, keepdim=False, dtype='int64', name=None)
其中keepdim(bool,可选)- 是否在输出Tensor中保留减小的维度。如果 keepdim 为True,则输出Tensor和 x 具有相同的维度(减少的维度除外,减少的维度的大小为1),默认值为False。

tensorlayerx源码paddle.argmax(x, axis=None, dtype='int64')

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions