生活随笔
收集整理的這篇文章主要介紹了
tf.Module
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
-
通過繼承tf.Module代替object任何tf.Variable或tf.Module分配給對象的屬性的實(shí)例可以使用被收集variables ， trainable_variables或submodules屬性：
import tensorflow as tf class Dense(tf.Module):def __init__(self, in_features, out_features, name=None):super(Dense, self).__init__(name=name)self.w = tf.Variable(tf.random.normal([in_features, out_features]), name='w')self.b = tf.Variable(tf.zeros([out_features]), name='b')def __call__(self, x):y = tf.matmul(x, self.w) + self.breturn tf.nn.relu(y)d = Dense(in_features=3, out_features=2)
d(tf.ones([1, 3]))
: <tf.Tensor: shape=(1, 2), dtype=float32, numpy=array([[0. , 0.5372393]], dtype=float32)>
總結(jié)
以上是生活随笔為你收集整理的tf.Module的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。
