theano_utils.py 文件源码

def make_node(self, A, b):
        A = T.as_tensor_variable(A)
        b = T.as_tensor_variable(b)
        assert A.ndim == 2
        assert b.ndim in [1, 2]
        otype = T.tensor(
            broadcastable=b.broadcastable,
            dtype=(A * b).dtype)
        return theano.gof.Apply(self, [A, b], [otype])