TensorFlow之矩阵变换

训练网络时,经常要对矩阵进行拼接拆分减少纬度扩充纬度改变shape转置乱序等操作,这里把常用到的方法总结归纳出来。

  1. tf.concat(values, axis, name="concat")
  2. tf.stack(values, axis=0, name="stack”)
  3. tf.unstack(value, num=None, axis=0, name="unstack”)
  4. tf.tile(input, multiples, name=None)
  5. tf.split(value, num_or_size_splits, axis=0, num=None, name="split”)
  6. tf.slice(input_, begin, size, name=None)
  7. tf.expand_dims(input, axis=None, name=None, dim=None)
  8. tf.squeeze(input, axis=None, name=None, squeeze_dims=None)
  9. tf.reshape(tensor, shape, name=None)
  10. tf.transpose(a, perm=None, name="transpose", conjugate=False)
  11. tf.random_shuffle(value, seed=None, name=None)
  1. tf.concat(values, axis, name=”concat”)

values中的矩阵沿着纬度axis拼接起来,纬度不变

a = tf.constant([[4, 2, 1], [1, 2, 3]], dtype=tf.float32)
b = tf.constant([[3, 2, 1], [0, 2, 1]], dtype=tf.float32)
c = tf.concat([a, b], axis=0)
c =
[[4. 2. 1.]
 [1. 2. 3.]
 [3. 2. 1.]
 [0. 2. 1.]]
  1. tf.stack(values, axis=0, name=”stack”)

values中的矩阵沿着纬度axis拼接起来,纬度+1,相比而言,tf.concat在网络中用的更多

a = tf.constant([[4, 2, 1], [1, 2, 3]], dtype=tf.float32)
b = tf.constant([[3, 2, 1], [0, 2, 1]], dtype=tf.float32)
c = tf.stack([a, b], axis=0)
c =
[[[4. 2. 1.]
  [1. 2. 3.]]

 [[3. 2. 1.]
  [0. 2. 1.]]]
  1. tf.unstack(value, num=None, axis=0, name=”unstack”)

value中的矩阵沿着纬度axis拆分出来,纬度-1

a = tf.constant([[4, 2, 1], [1, 2, 3]], dtype=tf.float32)
b = tf.constant([[3, 2, 1], [0, 2, 1]], dtype=tf.float32)
c = tf.stack([a, b], axis=0)
d1, d2 = tf.unstack(c, axis=0)
d1 =
[[4. 2. 1.]
 [1. 2. 3.]]
d2 =
[[3. 2. 1.]
 [0. 2. 1.]]
  1. tf.tile(input, multiples, name=None)

复制矩阵input,multiples中指定每个纬度上复制次数

a = tf.constant([[1, 2], [3, 4]], dtype=tf.float32)
b = tf.tile(a, [2, 2])
b =
[[1. 2. 1. 2.]
 [3. 4. 3. 4.]
 [1. 2. 1. 2.]
 [3. 4. 3. 4.]]
  1. tf.split(value, num_or_size_splits, axis=0, num=None, name=”split”)

把矩阵value,沿着axis纬度拆分成num_or_size_splits个小矩阵

a = tf.constant([[1, 2], [3, 4], [3, 5]], dtype=tf.float32)
b1, b2, b3 = tf.split(a, num_or_size_splits=3, axis=0)
b1 =
[[1. 2.]]
b2 =
[[3. 4.]]
b3 =
[[3. 5.]]
  1. tf.slice(input_, begin, size, name=None)

把矩阵input_,沿着每个纬度指定开始位置begin截取size大小的内容,这里-1代表到当前纬度结尾处,和python中list分块[begin:end]是类似的

a = tf.constant([[1, 2], [3, 4], [3, 5]], dtype=tf.float32)
b = tf.slice(a, [1, 0], [2, -1])
b =
[[3. 4.]
 [3. 5.]]
  1. tf.expand_dims(input, axis=None, name=None, dim=None)

把矩阵input的纬度增加1,axis为曾加的纬度位置,例如当前a的shape为[2],最后一个位置增加一个纬度shape变成[2,1]

a = tf.constant([1, 2], dtype=tf.float32)
b = tf.expand_dims(a, -1)
b =
[[1.]
 [2.]]
  1. squeeze(input, axis=None, name=None, squeeze_dims=None)
    把矩阵inputsize大小为1的纬度减去,axis为减少的纬度位置,例如当前a的shape为[2,1],最后一个位置减去一个纬度shape变成[2]
a = tf.constant([[1], [2]], dtype=tf.float32)
b = tf.squeeze(a, 1)
b =
[1. 2.]
  1. tf.reshape(tensor, shape, name=None)
    把矩阵tensor改变成指定shape的样子,需要注意的是,改变shape之前和之后的元素个数总和应该一样
a = tf.constant([[1, 2, 3, 4]], dtype=tf.float32)
b = tf.reshape(a, [2, 2])
b =
[[1. 2.]
 [3. 4.]]
  1. tf.transpose(a, perm=None, name=”transpose”, conjugate=False)
    调整矩阵a纬度的顺序,按照perm中指定顺序排列,列如把a的行列交换顺序,也就是经常所见的二维矩阵转置过程
a = tf.constant([[1, 2], [3, 4], [3, 5]], dtype=tf.float32)
b = tf.transpose(a, perm=[1, 0])
b =
[[1. 3. 3.]
 [2. 4. 5.]]
  1. tf.random_shuffle(value, seed=None, name=None)
    把矩阵value中的元素顺序打乱,需要注意的是被打乱顺序的纬度索引为0
a = tf.constant([[1, 2], [3, 4], [3, 5]], dtype=tf.float32)
b = tf.random_shuffle(a)
b =
[[3. 4.]
 [1. 2.]
 [3. 5.]]
    原文作者:看_有灰碟
    原文地址: https://www.jianshu.com/p/6b0815cd1326
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞