tf.reverse(
tensor, # 需要进行反转的张量,类型必须为其中的一个uint8, int8, uint16, int16, int32, int64, bool, bfloat16, half, float32, float64, complex64, complex128, string。最高维度为8-D
axis, # 需要反转的axis,注意axis的范围是[-rank(tensor),rank(tensor))之间。
name=None #可选的
)
import tensorflow as tf
import numpy as np
t=[[1,2,3],[4,5,6]]
t0=tf.reverse(t,[0])
t1=tf.reverse(t,[1])
t21=tf.reverse(t,[0,1])
t22=tf.reverse(t,[1,0])
sess=tf.Session()
print(sess.run(t0))
print(sess.run(t1))
print(sess.run(t21))
print(sess.run(t22))
#输出结果
#按行(axis=0)变换,上下行交换位置
[[4 5 6]
[1 2 3]]
#axis=1按照横轴变换,前后列交换位置
[[3 2 1]
[6 5 4]]
#先纵轴变换,后横轴变换
[[6 5 4]
[3 2 1]]
#先横轴后纵轴变换
[[6 5 4]
[3 2 1]]
本文详细介绍了 TensorFlow 中的 tf.reverse 函数,展示了如何通过该函数实现张量在不同轴上的翻转操作,并提供了具体的代码示例及运行结果,有助于读者理解 tf.reverse 的使用方法。
38

被折叠的 条评论
为什么被折叠?



