Sha256: bb73f581f3fca0cc70965b39a4274662acee3b2a9b5f208f2bd8f745bb98b637
Contents?: true
Size: 338 Bytes
Versions: 56
Compression:
Stored size: 338 Bytes
Contents
module Torch
module NN
class PairwiseDistance < Module
def initialize(p: 2.0, eps: 1e-6, keepdim: false)
super()
@norm = p
@eps = eps
@keepdim = keepdim
end
def forward(x1, x2)
F.pairwise_distance(x1, x2, p: @norm, eps: @eps, keepdim: @keepdim)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems