派托奇的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 ,
1