Tensorflow常用函数

古城微笑少年丶 2022-03-16 04:22 304阅读 0赞

# 基础函数 #

**（1）tf.add\_to\_collection() **

tf.add\_to\_collection（name,value），作用是将value以name的名字存储在一个是收集器中。在正则化计算中，常常用做：tf.add\_to\_collection('losses',tf.contrib.layers.l2\_regularizer(regularizer)(w))，就是讲正则化计算结果放入“losses”中

**（2）tf.get\_collection() **

tf.get\_collection(key)，作用是可以取出收集器中存入的内容，例如tf.get\_collection("losses")，就是取出以“losses”为key的存储内容，生成一个列表

**（3）tf.add\_n()**

看下面的一段代码

import tensorflow as tf
    import os
    
    os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"  # 忽略tensorflow警告信息
    import tensorflow as tf
    tf.add_to_collection('losses', tf.constant(2.2))
    tf.add_to_collection('losses', tf.constant(2.))
    tf.add_to_collection('losses', tf.constant(3.))
    with tf.Session() as sess:
        print(sess.run(tf.get_collection('losses')))
        print(sess.run(tf.add_n(tf.get_collection('losses'))))

输出的结果：

> \[2.2, 2.0, 3.0\]  
> 7.2

**（4）tf.argmax(x,axis)**

返回最大值所在的索引号，例如tf.argmax(\[1,0,0\],1)，返回0

**（5）tf.cast(x,dtype)**

把x转为dtype类型

**（6）tf.assign()**

函数原型 def assign(ref,value)

功能：将value的值赋值给ref。

需要注意，ref必须是通过tf.Variable()创建的tensor，如果ref = tf.constant()，会报错！

（7）tf.constant() \# 创建一个常量

（8）tf.matmul(m1,m2) \# 矩阵的乘法

（9）tf.Variable(\[1,2\])  \# 创建一个变量，注意变量在使用前，必须进行初始化

（10） tf.subtract( )  \# 矩阵的减法

（11） tf.add( ) \# 矩阵的加法

发表评论取消回复

表情：

评论列表（有 0 条评论，304人围观）

还没有评论，来说两句吧...

相关阅读

相关 TensorFlow常用操作

指定GPU 终端指定 CUDA_VISIBLE_DEVICES=0 nohup python demo.py >> base_log.out 2>&1 &

素颜马尾好姑娘i/ 2022年09月08日 00:09/ 0 赞/ 143 阅读

相关 tensorflow常用函数及概念

命令式编程与声明式编程命令式编程(imperative programming)：每个语句都按原来的意思执行，可以精确控制行为。通常可以无缝的和主语言交互，方便的利用主

迈不过友情╰/ 2022年06月17日 03:22/ 0 赞/ 222 阅读

相关 TensorFlow中常用函数说明

本文主要对TensorFlow的一些常用概念与方法进行描述，TensorFlow常用函数如下：转载地址：[https://blog.csdn.net/lenbow/arti

忘是亡心i/ 2022年05月27日 09:55/ 0 赞/ 234 阅读

相关 tensorflow常用函数接口汇总以及用tensorflow实现自编码机模型

最近学习tensorflow，这里做一个小总结，先说一下我的运行环境，win10+Anaconda3.5，导入tensorflow。首先提供一个trick，对于一些初次见到

痛定思痛。/ 2022年05月26日 08:35/ 0 赞/ 212 阅读

相关 Tensorflow常用函数

源文： https://www.w3cschool.cn/tensorflow\_python/tensorflow\_python-s62t2d4y.html

我会带着你远行/ 2022年05月20日 05:19/ 0 赞/ 188 阅读

相关 TensorFlow学习（三）之-——常用随机函数用法与numpy常用随机函数

一、TensorFlow常用随机函数 1.tf.constant tf.constant(value, dtype=None, shape=None)

冷不防/ 2022年05月18日 00:04/ 0 赞/ 234 阅读

相关 [keras] tensorflow backend 常用函数

`K.tile(input_tensor , tile_tensor)`对tensor进行复制同`tf.tile()` K.tile(inputs,[1,1,1

喜欢ヅ旅行/ 2022年05月13日 05:40/ 0 赞/ 278 阅读

相关 Tensorflow常用函数笔记

tf.concat 把一组向量从某一维上拼接起来，很向numpy中的Concatenate，官网例子： t1 = \[\[1, 2, 3\], \[4, 5, 6\]\]

淩亂°似流年/ 2022年04月15日 00:11/ 0 赞/ 264 阅读

相关 Tensorflow常用函数

基础函数（1）tf.add\_to\_collection() tf.add\_to\_collection（name,value），作用是将value以name的

古城微笑少年丶/ 2022年03月16日 04:22/ 0 赞/ 305 阅读

相关 TensorFlow 和 Numpy 等常用函数小结

1.tf.cast用法 tf.cast：用于改变某个张量的数据类型 import tensorflow as tf; import numpy as n

曾经终败给现在/ 2021年11月05日 13:36/ 0 赞/ 309 阅读