Pytorch中nn.模块中的forward函数可以替换为Tensorflow中tf.模块中的"__call__()"函数或keras中tf. keras. layers. layer中的call()函数。这是一个简单的tensorflow和keras密集层的例子：
tensorflow ：

class Dense(tf.Module):
  def __init__(self, input_dim, output_size, name=None):
     super().__init__(name=name)
     self.w = tf.Variable(tf.random.normal([input_dim, output_size]), name='w')
     self.b = tf.Variable(tf.zeros([output_size]), name='b')
  def __call__(self, x):
     y = tf.matmul(x, self.w) + self.b
     return tf.nn.relu(y)

凯拉斯：

class Dense(tf.keras.Layers.Layer):
  def __init__(self, units=32):
     super(SimpleDense, self).__init__()
     self.units = units
  def build(self, input_shape):
     self.w = self.add_weight(shape=(input_shape[-1], self.units),
                           initializer='random_normal',
                           trainable=True)
     self.b = self.add_weight(shape=(self.units,),
                           initializer='random_normal',
                           trainable=True)
  def call(self, inputs):
     return tf.matmul(inputs, self.w) + self.b

您可以查看以下链接以了解更多详细信息：

1. https://www.tensorflow.org/api_docs/python/tf/Module
2. https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer