深度学习TF—2.TensorFlow2高阶操作

最新推荐文章于 2025-03-04 13:32:48 发布

哎呦-_-不错

最新推荐文章于 2025-03-04 13:32:48 发布

阅读量648

点赞数

分类专栏： # 深度学习-Tensorflow 文章标签： tensorflow2

本BLOG上原创文章未经本人许可，不得用于商业用途，转载请注明出处。

本文链接：https://blog.youkuaiyun.com/weixin_46649052/article/details/113931703

版权

本文详细介绍了TensorFlow2中的高阶操作，包括张量的合并与分割、数据统计、排序、填充与复制、限幅以及高阶操作。讲解了如tf.concat、tf.stack、tf.norm、tf.reduce_max/min/mean/sum、tf.argsort、tf.sort、tf.pad、tf.tile、tf.clip_by_value等关键函数的用法，是理解TensorFlow2高级特性的宝贵资料。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

一、合并与分割

1. tf.concat-合并-原有的维度上进行累加

concat操作需要满足除拼接维度外，其余维度均相等
在这里插入图片描述

2.tf.stack-合并-创造一个新的维度

stack操作需要Tensor维度均相等
在这里插入图片描述

3.tf.unstack-分割

不能指定打散的数量，只能按维度进行分割
在这里插入图片描述

4.tf.split-分割

可以指定打散的数量
在这里插入图片描述

二、数据统计操作

1.tf.norm—张量的范数

二范数
在这里插入图片描述
一范数

2.tf.reduce_max/min/mean/sum—张量的最大值、最小值、平均值、和

在这里插入图片描述

3.tf.argmax/argmin—张量最大值的位置与最小值的位置

在这里插入图片描述

4.tf.equal—张量的比较

在这里插入图片描述

5.tf.unique—张量的独特值

在这里插入图片描述

三、张量排序

1.tf.sort-排序/tf.argsort-排序并返回索引

在这里插入图片描述

2.tf.math.top_k-最大值的前几个

在这里插入图片描述

3.案例

# 将无关信息屏蔽掉
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf
tf.random.set_seed(2467)

# output->[b,n] target->[b,]
def accuracy(output,target,topk=(1,)):
    maxk = max(topk)
    batch_size = target.shape[0]

    # 返回最大值前maxk个的索引
    pred = tf.math.top_k(output,maxk).indices
    # 转置
    pred = tf.transpose(pred,perm=[1,0])
    # 将target广播成pred形状
    target_ = tf.broadcast_to(target,pred.shape)
    # 比较
    correct = tf.equal(pred,target_)

    res = []
    for k in topk:
        correct_k = tf.cast(tf.reshape(correct[:k],[-1]),dtype=tf.float32)
        # print('123=',correct_k)
        correct_k = tf.reduce_sum(correct_k)
        acc = float(correct_k / batch_size)
        res.append(acc)

    return res

if __name__ == '__main__':
    # 正态分布
    output = tf.random.normal([10,6])
    # 使6类概率总和为1
    output = tf.math.softmax(output,axis=1)
    # 均匀分布
    target = tf.random.uniform([10],maxval=6,dtype=tf.int32)
    print('prob:',output.numpy())
    pred = tf.argmax(output,axis=1)

最低0.47元/天解锁文章