homesampling

View code on Github

#

කෑදරනියැදීම

මෙන්නඅපි පිවිසුම් බෙදා හැරීමෙන් බොහෝ දුරට ටෝකනය නියැදි කරමු.

මෙන්නමෙම නියැදි ශිල්පීය ක්රම භාවිතා කරන අත්හදා බැලීමක් .

14importtorch1516fromlabml\_nn.samplingimportSampler

#

19classGreedySampler(Sampler):

#

පිවිසුම්බෙදා හැරීමෙන් බොහෝ දුරට ටෝකනය නියැදි කරන්න

20def\_\_call\_\_(self,logits:torch.Tensor):

#

24returnlogits.argmax(dim=-1)

Trending Research Paperslabml.ai