blob: f0156fa1a515dfaeef0092ec3b5f8060b325a785 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
import theano
from blocks.graph import ComputationGraph
class Apply(object):
def __init__(self, outputs, return_vars, stream):
if not isinstance(outputs, list):
outputs = [outputs]
if not isinstance(return_vars, list):
return_vars = [return_vars]
self.outputs = outputs
self.return_vars = return_vars
self.stream = stream
cg = ComputationGraph(self.outputs)
self.input_names = [i.name for i in cg.inputs]
self.f = theano.function(inputs=cg.inputs, outputs=self.outputs)
def __iter__(self):
self.iterator = self.stream.get_epoch_iterator(as_dict=True)
while True:
try:
batch = next(self.iterator)
except StopIteration:
return
inputs = [batch[n] for n in self.input_names]
outputs = self.f(*inputs)
def find_retvar(name):
for idx, ov in enumerate(self.outputs):
if ov.name == name:
return outputs[idx]
if name in batch:
return batch[name]
raise ValueError('Variable ' + name + ' neither in outputs or in batch variables.')
yield {name: find_retvar(name) for name in self.return_vars}
|
