生活随笔
收集整理的這篇文章主要介紹了
tf.map_fn
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
import numpy as np
import tensorflow as tf
elems = np.array([1, 2, 3, 4, 5, 6])
tf.map_fn(lambda x: x * x, elems)
<tf.Tensor: shape=(6,), dtype=int64, numpy=array([ 1, 4, 9, 16, 25, 36])>
總結(jié)
以上是生活随笔為你收集整理的tf.map_fn的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。
