Sha256: a9167854336fa96c6010e01ba3505cdc57d1fbe9a1b05daa1d405ceaf169ad90

Contents?: true

Size: 456 Bytes

Versions: 56

Compression:

Stored size: 456 Bytes

Contents

module Torch
  module NN
    class CTCLoss < Loss
      def initialize(blank: 0, reduction: "mean", zero_infinity: false)
        super(reduction)
        @blank = blank
        @zero_infinity = zero_infinity
      end

      def forward(log_probs, targets, input_lengths, target_lengths)
        F.ctc_loss(log_probs, targets, input_lengths, target_lengths, blank: @blank, reduction: @reduction, zero_infinity: @zero_infinity)
      end
    end
  end
end

Version data entries

56 entries across 56 versions & 1 rubygems

Version Path
torch-rb-0.18.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.17.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.17.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.16.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.15.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.14.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.14.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.13.2 lib/torch/nn/ctc_loss.rb
torch-rb-0.13.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.13.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.12.2 lib/torch/nn/ctc_loss.rb
torch-rb-0.12.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.12.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.11.2 lib/torch/nn/ctc_loss.rb
torch-rb-0.11.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.11.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.10.2 lib/torch/nn/ctc_loss.rb
torch-rb-0.10.1 lib/torch/nn/ctc_loss.rb
torch-rb-0.10.0 lib/torch/nn/ctc_loss.rb
torch-rb-0.9.2 lib/torch/nn/ctc_loss.rb