tf.argmax(vector, 1):返回的是vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。

tesorflow函数

输出:

[4]
[2 1]

 

tf.clip_by_value的用法

该函数主要是为了防止,gradiant计算得到的值太大或者太小

tf.clip_by_value(A, min, max):输入一个张量A,把A中的每一个元素的值都压缩在min和max之间。小于min的让它等于min,大于max的元素的值等于max。

tesorflow函数

输出结果:

[[2 2 3 3]
 [3 3 3 2]]

 

tf.expand_dims() 函数

tf.expand_dims(input, axis=None, name=None, dim=None)

函数为在第axis位置上增加一个维度,给定张量输入,此操作在输入形状的维度索引轴处插入1的尺寸,尺寸索引轴从零开始,如果指定的轴为负数,则从后向前计数。

tesorflow函数

 tf.squeeze()

tf.squeeze(input, squeeze_dims=None, name=None)

从input的tensor中删除所有大小为1的维度,如果不想删除所有尺寸1尺度,可以通过指定dims来删除特定尺寸1尺寸的维度。

如果某维度为1就降掉这一维。
当前维度是(2, 3, 1, 4) => (2, 3, 4)
当前维度是(1, 2, 1, 4) => (2, 4)
与squeeze函数相反的是expand_dims函数。

 

相关文章:

  • 2022-12-23
  • 2021-08-18
  • 2021-11-05
  • 2022-12-23
  • 2021-11-26
  • 2021-12-20
  • 2022-12-23
  • 2021-04-05
猜你喜欢
  • 2021-11-04
  • 2021-04-13
  • 2021-11-26
  • 2021-09-03
  • 2022-12-23
  • 2021-08-14
  • 2022-12-23
相关资源
相似解决方案