Esempio n. 1
0
	def _transmit(self, ctx):
		if ctx.trace is not None:
			array.softmax(ctx.trace.odata, ctx.src)
			if ctx.dst is not None:
				array.copy(ctx.dst, ctx.trace.odata)
		else:
			Softmax._transmit(self, ctx)
Esempio n. 2
0
	def __init__(self, size, **kwargs):
		Loss.__init__(self)
		Softmax.__init__(self, size, **kwargs)