TorchLean API

Docs HomeInstallationGuideExamplesGraphsTools

NN.Runtime.Autograd.Torch.Core.Session.References

Session References #

Tensor and integer handles carry both a session owner and a recording generation. Check those before using an id: resetting a tape can reuse the same numeric index.

source
def Runtime.Autograd.Torch.Internal.EagerSession.currentRefIdentity {α : Type} [TorchLean.Storage α] (s : EagerSession α) :
IO RefIdentity

Capture the current owner and generation for a newly recorded handle.

Instances For
    source
    def Runtime.Autograd.Torch.Internal.EagerSession.makeTensorRef {α : Type} [TorchLean.Storage α] {sh : Spec.Shape} (s : EagerSession α) (id : ℕ) :
    IO (TensorRef α sh)

    Construct a tensor handle owned by the current recording phase.

    Instances For
      source
      def Runtime.Autograd.Torch.Internal.EagerSession.makeNatRef {α : Type} [TorchLean.Storage α] (s : EagerSession α) (id : ℕ) :
      IO NatRef

      Construct a non-differentiable handle owned by the current recording phase.

      Instances For
        source
        def Runtime.Autograd.Torch.Internal.EagerSession.validateTensorRef {α : Type} [TorchLean.Storage α] (s : EagerSession α) {sh : Spec.Shape} (x : TensorRef α sh) :
        IO Unit

        Validate one tensor handle before using its numeric tape id.

        Instances For
          source
          def Runtime.Autograd.Torch.Internal.EagerSession.validateRefIdentities {α : Type} [TorchLean.Storage α] (s : EagerSession α) (identities : Array (Option RefIdentity)) :
          IO Unit

          Validate tensor handles consumed by one operation.

          Instances For
            source
            def Runtime.Autograd.Torch.Internal.EagerSession.validateNatRef {α : Type} [TorchLean.Storage α] (s : EagerSession α) (x : NatRef) :
            IO Unit

            Validate one non-differentiable handle before using its environment index.

            Instances For
              source
              def Runtime.Autograd.Torch.Internal.EagerSession.inputNat {α : Type} [TorchLean.Storage α] (s : EagerSession α) (v : ℕ) :
              IO NatRef

              Record a non-differentiable Nat input in the session environment.

              This supports ops that depend on indices/labels that should not receive gradients.

              Instances For
                source
                def Runtime.Autograd.Torch.Internal.EagerSession.getNat {α : Type} [TorchLean.Storage α] (s : EagerSession α) (r : NatRef) :
                IO ℕ

                Read a previously recorded NatRef.

                Instances For
                  source
                  def Runtime.Autograd.Torch.Internal.EagerSession.setNat {α : Type} [TorchLean.Storage α] (s : EagerSession α) (r : NatRef) (v : ℕ) :
                  IO Unit

                  Overwrite a previously recorded NatRef.

                  Instances For