Function softmax

utils/__init__.py:160–161 · view source on GitHub ↗

(x, dim)

Source from the content-addressed store, hash-verified

158
159
160	def softmax(x, dim):
161	return F.softmax(x, dim=dim, dtype=torch.float32)
162
163
164	def unpack_dict_to_list(samples):

nothing calls this directly

no outgoing calls

no test coverage detected