擅长:python、mysql、java
<p>我能想到的最好的办法是:</p>
<pre><code>import tensorflow as tf
inp = tf.reshape(tf.range(1, 11), (2, 5))
inputlayer = tf.keras.layers.Input(shape=(5,))
out = tf.keras.layers.Lambda(lambda x: tf.map_fn(lambda y:
tf.random.shuffle(y), x))(inputlayer)
model = tf.keras.Model(inputlayer, out)
model(inp)
</code></pre>
<p>由此:</p>
<pre><code><tf.Tensor: shape=(2, 5), dtype=int32, numpy=
array([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10]])>
</code></pre>
<p>为此:</p>
<pre><code><tf.Tensor: shape=(2, 5), dtype=float32, numpy=
array([[ 4., 1., 5., 3., 2.],
[10., 6., 9., 8., 7.]], dtype=float32)>
</code></pre>