
package com.intel.analytics.bigdl.nn

import com.intel.analytics.bigdl.tensor.Tensor
import com.intel.analytics.bigdl.utils.serializer.ModuleSerializationTest

import scala.util.Random

class SoftMinSerialTest extends ModuleSerializationTest {
  override def test(): Unit = {
    val softMin = SoftMin[Float]().setName("softMin")
    val input = Tensor[Float](10).apply1(_ => Random.nextFloat())
    runSerializationTest(softMin, input)
  }
}
