com.kotlinnlp.simplednn.deeplearning.birnn.BiRNNEncodersPool.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of simplednn Show documentation

SimpleDNN is a machine learning lightweight open-source library written in Kotlin whose purpose is to support the development of feed-forward and recurrent Artificial Neural Networks.

There is a newer version: 0.14.0

Show newest version

/* Copyright 2016-present The KotlinNLP Authors. All Rights Reserved.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, you can obtain one at http://mozilla.org/MPL/2.0/.
 * ------------------------------------------------------------------*/

package com.kotlinnlp.simplednn.deeplearning.birnn

import com.kotlinnlp.simplednn.simplemath.ndarray.NDArray
import com.kotlinnlp.simplednn.utils.ItemsPool

/**
 * A pool of [BiRNNEncoder]s which allows to allocate and release one when needed, without creating a new one.
 * It is useful to optimize the creation of new structures every time a new encoder is created.
 *
 * @property birnn the [BiRNN] which the encoders of the pool will work with
 */
class BiRNNEncodersPool>(
  val birnn: BiRNN
) : ItemsPool>() {

  /**
   * The factory of a new [BiRNNEncoder].
   *
   * @param id the unique id of the item to create
   *
   * @return a new [BiRNNEncoder] with the given [id]
   */
  override fun itemFactory(id: Int) = BiRNNEncoder(network = this.birnn, id = id)
}