在Tensorflow 2中定义自己的焊盘

import tensorflow as tf
from tensorflow.keras.layers import Layer

class WeldingLayer(Layer):
    def __init__(self, num_outputs, **kwargs):
        super(WeldingLayer, self).__init__(**kwargs)
        self.num_outputs = num_outputs

    def build(self, input_shape):
        self.kernel = self.add_weight("kernel", shape=[input_shape[-1], self.num_outputs])

    def call(self, inputs):
        output = tf.matmul(inputs, self.kernel)
        return output

# 使用自定义层
input_data = tf.random.normal(shape=(32, 64)) # 输入数据
welding_layer = WeldingLayer(num_outputs=128)
output_data = welding_layer(input_data) # 输出数据