siren pytorch:具有周期激活函数的SIREN 隐式神经表示的Pytorch实现。 源码

派托奇的SIREN SIREN-Pytorch实现。 安装 $ pip install siren-pytorch 用法 基于SIREN的多层神经网络 import torch from torch import nn from siren_pytorch import SirenNet net = SirenNet ( dim_in = 2 , # input dimension, ex. 2d coor dim_hidden = 256 , # hidden dimension dim_out = 3 , # output dimension, ex. rgb value num_layers = 5 ,