#

參數(shù)層

  • Pytorch 實(shí)現(xiàn)自定義參數(shù)層的例子

    注意，一般官方接口都帶有可導(dǎo)功能，如果你實(shí)現(xiàn)的層不具有可導(dǎo)功能，就需要自己實(shí)現(xiàn)梯度的反向傳遞。 官方Linear層： class Linear(Module): def __init__(se

    作者：青盞
    2020-09-11 06:24:01