Tensorflow初学基本操作

1.从tensor里截取特定的行:

比如有temp_var = tf.Variable([[1,2,3], [4,5,6], [7,8,9], [10,11,12]])

想要取出来的行为idx = tf.constant([0,2])

那么相应的操作为:rows = tf.gather(temp_var, idx)

gather_nd 比gather还要更强势

2.tf.nn.top_k()

def top_k(input, k=1, sorted=True, name=None)

查找最后一个维度的前k 个最大条目的值和索引。

注:该函数返回的数据包含两部分,第一部分是返回的value值,第二部分返回的是对应的索引值。可以通过索引[0]或者[1]进行访问。

3.命名习惯

养成在tf.xx()里面加上 name = xx的习惯!


4.恩...expand_dim 在干嘛

t0=tf.constant([[0,0],[1,1]])
t1 = tf.expand_dims(t0, axis=-1)
t=tf.constant([[[0],[0]],[[1],[1]]])

5. 程序的话最好还是尽量避免for循环 学会逼自己用tensor matrix进行矩阵之间的计算

这样的话更新啥的还是比较方便 分成小的再凑起来总会是有问题的

6. tf.concat 和tf.stack 用作将一组tensor合并(已有dimension vs 新增dimension) 和reduce_mean可能配合使用

7. 乘以不同阶数的矩阵tensor 记得reshape 然后再相乘

8. tf.nn.embedding_lookup_sparse 和 tf.sparse.matmul(sparse_tensor_dense_matmul)之间的瓜葛

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值