Packages

class AMSGrad extends Optimizer

Linear Supertypes
Known Subclasses
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. AMSGrad
  2. Optimizer
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Instance Constructors

  1. new AMSGrad(learningRate: Float = 0.001f, decay: Schedule[Float] = FixedSchedule[Float](), beta1: Float = 0.9f, beta2: Float = 0.999f, epsilon: Float = 1e-8f, useLocking: Boolean = false, learningRateSummaryTag: String = null, name: String = "AMSGrad")
    Attributes
    protected

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##: Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. def applyDense[T, I](gradient: Output[T], variable: variables.Variable[T], iteration: Option[variables.Variable[I]])(implicit arg0: core.types.TF[T], arg1: core.types.IsNotQuantized[T], arg2: core.types.TF[I], arg3: core.types.IsIntOrLong[I]): UntypedOp
    Definition Classes
    AMSGradOptimizer
  5. def applyGradients[T, I](gradientsAndVariables: Seq[(OutputLike[T], variables.Variable[Any])], iteration: Option[variables.Variable[I]] = None, name: String = this.name)(implicit arg0: core.types.TF[T], arg1: LongDefault[I], arg2: core.types.TF[I], arg3: core.types.IsIntOrLong[I]): UntypedOp
    Definition Classes
    Optimizer
    Annotations
    @throws(scala.this.throws.<init>$default$1[IllegalArgumentException])
  6. def applySparse[T, I](gradient: OutputIndexedSlices[T], variable: variables.Variable[T], iteration: Option[variables.Variable[I]])(implicit arg0: core.types.TF[T], arg1: core.types.IsNotQuantized[T], arg2: core.types.TF[I], arg3: core.types.IsIntOrLong[I]): UntypedOp
    Definition Classes
    AMSGradOptimizer
  7. def applySparseDuplicateIndices[T, I](gradient: OutputIndexedSlices[T], variable: variables.Variable[T], iteration: Option[variables.Variable[I]])(implicit arg0: core.types.TF[T], arg1: core.types.IsNotQuantized[T], arg2: core.types.TF[I], arg3: core.types.IsIntOrLong[I]): UntypedOp
    Definition Classes
    Optimizer
  8. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  9. val beta1: Float
  10. var beta1Tensor: Output[Float]
    Attributes
    protected
  11. val beta2: Float
  12. var beta2Tensor: Output[Float]
    Attributes
    protected
  13. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.CloneNotSupportedException]) @native() @HotSpotIntrinsicCandidate()
  14. def computeGradients[T](loss: Output[T], lossGradients: Seq[OutputLike[T]] = null, variables: Set[variables.Variable[Any]] = null, gradientsGatingMethod: GatingMethod = Gradients.OpGating, gradientsAggregationMethod: AggregationMethod = Gradients.AddAggregationMethod, colocateGradientsWithOps: Boolean = false)(implicit arg0: core.types.TF[T], arg1: core.types.IsFloatOrDouble[T]): Seq[(OutputLike[T], variables.Variable[Any])]
    Definition Classes
    Optimizer
    Annotations
    @throws(scala.this.throws.<init>$default$1[IllegalArgumentException])
  15. def createSlots(variables: Seq[variables.Variable[Any]]): Unit
    Definition Classes
    AMSGradOptimizer
  16. val decay: Schedule[Float]
  17. val epsilon: Float
  18. var epsilonTensor: Output[Float]
    Attributes
    protected
  19. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  20. def equals(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef → Any
  21. def finish(updateOps: Set[UntypedOp], nameScope: String): UntypedOp
    Definition Classes
    AMSGradOptimizer
  22. def getBeta1[V](variable: variables.Variable[V])(implicit arg0: core.types.TF[V]): Output[V]
    Attributes
    protected
  23. def getBeta2[V](variable: variables.Variable[V])(implicit arg0: core.types.TF[V]): Output[V]
    Attributes
    protected
  24. def getBetaPowerAccumulators: (variables.Variable[Float], variables.Variable[Float])
    Attributes
    protected
  25. final def getClass(): Class[_ <: AnyRef]
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @HotSpotIntrinsicCandidate()
  26. def getEpsilon[V](variable: variables.Variable[V])(implicit arg0: core.types.TF[V]): Output[V]
    Attributes
    protected
  27. def getLearningRate[V, I](variable: variables.Variable[V], iteration: Option[variables.Variable[I]])(implicit arg0: core.types.TF[V], arg1: core.types.TF[I], arg2: core.types.IsIntOrLong[I]): Output[V]
    Attributes
    protected
  28. final def getNonSlotVariable[T](name: String, graph: core.Graph = null): variables.Variable[T]
    Attributes
    protected
    Definition Classes
    Optimizer
  29. final def getNonSlotVariables: Iterable[variables.Variable[Any]]
    Attributes
    protected
    Definition Classes
    Optimizer
  30. final def getOrCreateNonSlotVariable[T](name: String, initialValue: tensors.Tensor[T], colocationOps: Set[UntypedOp] = Set.empty, ignoreExisting: Boolean = false)(implicit arg0: core.types.TF[T]): variables.Variable[T]
    Attributes
    protected
    Definition Classes
    Optimizer
  31. final def getSlot[T, R](name: String, variable: variables.Variable[T])(implicit arg0: core.types.TF[R]): variables.Variable[R]
    Attributes
    protected
    Definition Classes
    Optimizer
  32. final def getSlot[T, R](name: String, variable: variables.Variable[T], dataType: core.types.DataType[R], initializer: Initializer, shape: core.Shape, variableScope: String)(implicit arg0: core.types.TF[R]): variables.Variable[R]
    Attributes
    protected
    Definition Classes
    Optimizer
  33. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @HotSpotIntrinsicCandidate()
  34. val ignoreDuplicateSparseIndices: Boolean
    Definition Classes
    AMSGradOptimizer
  35. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  36. val learningRate: Float
  37. val learningRateSummaryTag: String
  38. var learningRateTensor: Output[Float]
    Attributes
    protected
  39. def minimize[T, I](loss: Output[T], lossGradients: Seq[OutputLike[T]] = null, variables: Set[variables.Variable[Any]] = null, gradientsGatingMethod: GatingMethod = Gradients.OpGating, gradientsAggregationMethod: AggregationMethod = Gradients.AddAggregationMethod, colocateGradientsWithOps: Boolean = false, iteration: Option[variables.Variable[I]] = None, name: String = "Minimize")(implicit arg0: core.types.TF[T], arg1: core.types.IsFloatOrDouble[T], arg2: LongDefault[I], arg3: core.types.TF[I], arg4: core.types.IsIntOrLong[I]): UntypedOp
    Definition Classes
    Optimizer
    Annotations
    @throws(scala.this.throws.<init>$default$1[IllegalArgumentException])
  40. val name: String
    Definition Classes
    AMSGradOptimizer
  41. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  42. final val nonSlotVariables: Map[(String, Option[core.Graph]), variables.Variable[Any]]
    Attributes
    protected
    Definition Classes
    Optimizer
  43. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @HotSpotIntrinsicCandidate()
  44. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @HotSpotIntrinsicCandidate()
  45. def prepare[I](iteration: Option[variables.Variable[I]])(implicit arg0: core.types.TF[I], arg1: core.types.IsIntOrLong[I]): Unit
    Definition Classes
    AMSGradOptimizer
  46. final def slotNames: Set[String]
    Attributes
    protected
    Definition Classes
    Optimizer
  47. final val slots: Map[String, Map[variables.Variable[Any], variables.Variable[Any]]]
    Attributes
    protected
    Definition Classes
    Optimizer
  48. final def state: Seq[variables.Variable[Any]]
    Definition Classes
    Optimizer
  49. final def synchronized[T0](arg0: => T0): T0
    Definition Classes
    AnyRef
  50. def toString(): String
    Definition Classes
    AnyRef → Any
  51. val useLocking: Boolean
    Definition Classes
    AMSGradOptimizer
  52. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  53. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException]) @native()
  54. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  55. final def zerosSlot[T](name: String, variable: variables.Variable[T], variableScope: String)(implicit arg0: core.types.TF[T]): variables.Variable[T]
    Attributes
    protected
    Definition Classes
    Optimizer

Deprecated Value Members

  1. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.Throwable]) @Deprecated @deprecated
    Deprecated

    (Since version ) see corresponding Javadoc for more information.

Inherited from Optimizer

Inherited from AnyRef

Inherited from Any

Ungrouped