本文整理汇总了Python中quagga.connector.Connector.assign_repeat方法的典型用法代码示例。如果您正苦于以下问题:Python Connector.assign_repeat方法的具体用法?Python Connector.assign_repeat怎么用?Python Connector.assign_repeat使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类quagga.connector.Connector
的用法示例。
在下文中一共展示了Connector.assign_repeat方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: RepeatBlock
# 需要导入模块: from quagga.connector import Connector [as 别名]
# 或者: from quagga.connector.Connector import assign_repeat [as 别名]
class RepeatBlock(object):
def __init__(self, x, repeats, axis=None, device_id=None):
self.context = Context(device_id)
device_id = self.context.device_id
self.repeats = repeats
self.axis = axis
learning = x.bpropagable
if learning:
self.x, self.dL_dx = x.register_usage(device_id, device_id)
else:
self.x = x.register_usage(device_id)
if axis == 0:
self.output = Matrix.empty(x.nrows * repeats, x.ncols, x.dtype, device_id)
elif axis == 1:
self.output = Matrix.empty(x.nrows, x.ncols * repeats, x.dtype, device_id)
else:
raise ValueError('TODO')
self.output = Connector(self.output, device_id if learning else None)
def fprop(self):
self.output.assign_repeat(self.context, self.x, self.repeats, self.axis)
self.output.fprop()
def bprop(self):
if hasattr(self, 'dL_dx'):
self.dL_dx.add_repeat_derivative(self.context, self.output.backward_matrix, self.repeats, self.axis)