gpt4 book ai didi

Tensorflow Argmax : What's the difference between "axis" and "dimension" parameter?

转载 作者:行者123 更新时间:2023-12-05 00:52:45 25 4
gpt4 key购买 nike

当前版本的 Tensorflow Argmax 没有指定“轴”和“维度”参数之间的区别。这是official manual 中给出的唯一信息:

tf.argmax(input, axis=None, name=None, dimension=None) {#argmax}

Returns the index with the largest value across axes of a tensor.

Args:

input: A Tensor. Must be one of the following types: float32, float64, int64, int32, uint8, uint16, int16, int8, complex64, complex128, qint8, quint8, qint32, half.

axis: A Tensor. Must be one of the following types: int32, int64. int32, 0 <= axis < rank(input). Describes which axis of the input Tensor to reduce across. For vectors, use axis = 0.

name: A name for the operation (optional).

Returns:

A Tensor of type int64.

有人可以澄清吗?哪一个实际上是降维的?

最佳答案

TensorFlow 正在过渡到使用 axis而不是 dimension即将弃用:https://www.tensorflow.org/install/migration

关于Tensorflow Argmax : What's the difference between "axis" and "dimension" parameter?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42397591/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com