tensorflow2.x学习笔记八:自定义网络层、模型以及损失函数

一、自定义网络层layer

  • 继承tf.keras.layers.Layer
  • 使用tf.keras.layers.Lambda

下面这个例子就包含了以上两种形式:

import numpy as np
from tensorflow import keras

class CustomizedDenseLayer(keras.layers.Layer):
    def __init__(self, units, activation

你可能感兴趣的:(#,tensorflow,深度学习,python)