(tensor, init_funcs)
| 34 | """Initialize a GRU layer. """ |
| 35 | |
| 36 | def _concat_init(tensor, init_funcs): |
| 37 | (length, fan_out) = tensor.shape |
| 38 | fan_in = length // len(init_funcs) |
| 39 | |
| 40 | for (i, init_func) in enumerate(init_funcs): |
| 41 | init_func(tensor[i * fan_in : (i + 1) * fan_in, :]) |
| 42 | |
| 43 | def _inner_uniform(tensor): |
| 44 | fan_in = nn.init._calculate_correct_fan(tensor, "fan_in") |