Tensorflow Argmax:“轴”和“尺寸”参数之间有什么区别?

Dr_*_*zuś 4 neural-network argmax tensorflow

Tensorflow Argmax的当前版本未指定“轴”和“尺寸”参数之间的区别。这是官方手册中提供的唯一信息:

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.
Run Code Online (Sandbox Code Playgroud)

有人可以澄清吗?缩小的尺寸实际上是哪一个?

yue*_*ngz 5

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