Sha256: 75de077e9159c18cdddfa5f8103696b9b243f1084ab9af364e2688e06deb0027
Contents?: true
Size: 278 Bytes
Versions: 56
Compression:
Stored size: 278 Bytes
Contents
module Torch
module NN
class CosineSimilarity < Module
def initialize(dim: 1, eps: 1e-8)
super()
@dim = dim
@eps = eps
end
def forward(x1, x2)
F.cosine_similarity(x1, x2, dim: @dim, eps: @eps)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems