in fast_grad_conv.py [0:0]
def forward(ctx, weight_param, bias_param, batch_size=1):
ctx.save_for_backward(weight_param)
weight = weight_param.repeat(batch_size, 1, 1, 1)
if bias_param is not None:
bias = bias_param.repeat(batch_size)
return weight, bias