MiniFlow -- 4.Linear Transform

还记得以前的要给公式

上面的表述方式可以理解为和公式的方式，接下来我们使用矩阵的方式，将所有的x表示为矩阵X，所有的w为矩阵W，当然b也可以理解为一个向量

如果我们认为 Linear 节点是一个1个输入，就是对应上面公式i=1，但是要对应k个输出，上面的公司就不能用了，接下来我们就得用矩阵的方式表示
我们使用X表示一个 1x1矩阵

W就变成了1xk矩阵

根据行列式乘法规则，1x1 * 1xk -->1xk 因为mxn * nxk --> mxk

那我们怎么将n个输入，对应到k个输出呢
X是一个1xn矩阵，W是一个nxk矩阵，我们想要得到一个1xk的矩阵，b仍然是要一个1xk矩阵

这里n是输入特征数量，k是输出特征数量，经过计算，我们得到一个矩阵，这个矩阵与k个特征，这就将输入n个特征映射到输出k个特征
根据行列式规则，我们不能用XW表示，得到一个1xk矩阵，然后根据矩阵加法，加上b，得到一个1xk矩阵

但是在实践，在每一次前向运算中我们使用一个批次喂入网络，而不是上面的1xk，这个时候输入就是 m x n，m表示批次

根据行列式规则，我们会得到一个mxk矩阵，m是批次，k是输出特征数量，b就得用广播来计算了

因此可以用公式表示为

代码

nn.py

from miniflow import *
import numpy as np


if __name__ == "__main__":
    X, W, b = Input(), Input(), Input()
    f = Linear(X, W, b)
    # 2*2 行批次，列特征
    X_ = np.array([[-1., -2.], [-1, -2]])
    # 行特征，列输出特征数量
    W_ = np.array([[2., -3], [2., -3]])
    b_ = np.array([-3., -5])
    feed_dict = {X: X_, W: W_, b: b_}
    graph = topological_sort(feed_dict)
    output = forward_pass(f, graph)

    print(output)

miniflow.py

import numpy as np

class Node(object):
    def __init__(self, inbound_nodes=[]):
        # 输入节点列表 用户前向计算
        self.inbound_nodes = inbound_nodes
        # 输出节点列表 用户反向传播
        self.outbound_nodes = []
        # A calculated value 本节点的计算值
        self.value = 0
        # Add this node as an outbound node on its inputs.
        # 将本节点作为输入节点的输出节点
        for n in self.inbound_nodes:
            n.outbound_nodes.append(self)

    def forward(self):
        """
        Forward propagation 虚函数
        前向运算 基于 inbound_nodes 输入节点列表的 value 计算输入节点列表的输出值
        存储在 self.value
        :return:
        """
        raise NotImplemented


class Input(Node):
    def __init__(self):
        """
        Input node 没有 inbound nodes ，他是整个神经网络的开始
        因此不需要进行任何操作
        """
        # 输入节点列表为空
        Node.__init__(self)

    def forward(self, value=None):
        if value is not None:
            self.value = value



class Linear(Node):
    def __init__(self, X, W, b):
        # Notice the ordering of the inputs passed to the
        # Node constructor.
        Node.__init__(self, [X, W, b])

    def forward(self):
        X = self.inbound_nodes[0].value
        W = self.inbound_nodes[1].value
        b = self.inbound_nodes[2].value
        self.value = np.dot(X, W) + b



"""
Can you augment the Add class so that it accepts
any number of nodes as input?

Hint: this may be useful:
https://docs.python.org/3/tutorial/controlflow.html#unpacking-argument-lists
"""


class Add(Node):
    def __init__(self, *inputs):
        """
        You could access x and y in forward with
        self.inbound_nodes[0] (x) and self.inbound_nodes[1](y)
        这里我们要计算x和y的和，那么输入节点就是x，y，那么x，y 就是本届点的输入节点，也就是self.inbound_nodes[0]
        :param x: Node input
        :param y: Node input
        """
        Node.__init__(self, inputs)

    def forward(self):
        for n in self.inbound_nodes:
            self.value += n.value


def topological_sort(feed_dict):
    """
    我们将所有节点构成一个graphs，一个计算图
    :param feed_dict: 这个一个输入节点的字典，key 是Input node value 是初始化的值
    :return: 返回一个序列化的节点列表
    """
    # 提取所有输入节点
    input_nodes = [n for n in feed_dict.keys()]
    G = {}
    # 所有输入节点列表
    nodes = [n for n in input_nodes]
    # 这里将所有输入节点和输出节点作为G的key node， value也是一个字典，in记录key node的所有输入节点列表，out记录key node 的所有输出节点列表
    # 其中一个G的一个元素如下，node 可以是Input 也可以是Add， 如果node是Input 那么in为空，如果node是末尾节点，out为空
    # 如果node是中间节点，name in是输入节点列表， out就是输出节点
    #       |-in list         |-in list
    # node1 |           node2 |
    #       |-out list        |-out list

    while len(nodes) > 0:
        # 从头部弹出一个节点 node
        n = nodes.pop(0)
        if n not in G:
            G[n] = {"in": set(), "out": set()}
        for m in n.outbound_nodes:
            if m not in G:
                G[m] = {"in": set(), "out": set()}
            G[n]["out"].add(m)
            G[m]["in"].add(n)

    L = []
    # 无序不重复
    S = set(input_nodes)
    # 这里序列化L，L为 input input out
    while len(S) > 0:
        n = S.pop()
        # 给输入节点赋值，如果n节点是输入节点，就将赋值
        if isinstance(n, Input):
            n.value = feed_dict[n]
        # 将节点n加到L中
        L.append(n)
        # 检查当前的节点的输出节点，n 可能是 Input Add，或者隐藏层 最后一层的节点没有outbound_nodes
        for m in n.outbound_nodes:
            # 删除当前节点中的输出
            G[n]["out"].remove(m)
            # 删除输出节点中的输入
            G[m]['in'].remove(n)
            # 如果 G[m]["in"] 为空的时候说明输入都添加完了，将输出添加到最后
            if len(G[m]["in"]) == 0:
                S.add(m)
    return L


def forward_pass(output_node, sorted_nodes):
    """
    Performs a forward pass through a list of sorted nodes.
    :param output_node: A node in the graph, should be the output node(have no outgoing edges)
    :param sorted_nodes: A topologically sorted list of nodes.
    :return: the out
    """
    for n in sorted_nodes:
        n.forward()

    return output_node.value