Tensorflow中 tf.argmax()方法的用法

tf.argmax()方法用于计算张量tensor沿着某一维度的最大值的索引。

tf.math.argmax(
    input,
    axis=None,
    name=None,
    dimension=None,
    output_type=tf.int64)

第一个参数 input: 输入的需要处理的张量 tensor,必须是以下数据类型之一float32float64int32uint8int16int8complex64int64qint8quint8qint32bfloat16uint16complex128halfuint32uint64
第二个参数 axis: 指定的维(或轴),取值为0,代表延列取最大值索引,取值为1,表示延行取最大值索引,如果不指定,则默认为按照列计算,取值为2,3,4...是,则为多维函数的相应维度,以此类推
第三个参数 name: 操作的名称;
第四个参数 dimension:已弃用;
第五个参数 output_type:返回值的数据类型,是tf.int32, tf.int64之一。默认为

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值