spikingjelly.activation_based.neuron_kernel package

Module contents

class spikingjelly.activation_based.neuron_kernel.MultiStepIFNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(hard_reset: bool, dtype: str)

static create_bptt_kernel(sg_cuda_code_fun, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, v_threshold: float, v_reset: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq)

class spikingjelly.activation_based.neuron_kernel.MultiStepLIFNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(decay_input: bool, hard_reset: bool, dtype: str, kernel_name_prefix: str = 'LIFNode')

static create_bptt_kernel(sg_cuda_code_fun, decay_input: bool, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, decay_input: bool, tau: float, v_threshold: float, v_reset: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq)

class spikingjelly.activation_based.neuron_kernel.MultiStepParametricLIFNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(decay_input: bool, hard_reset: bool, dtype: str)

static create_bptt_kernel(sg_cuda_code_fun, decay_input: bool, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, reciprocal_tau: Tensor, decay_input: bool, v_threshold: float, v_reset: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq)

spikingjelly.activation_based.neuron_kernel.check_multi_step_neuron_output_and_grad(device, multi_step_neuron, shape=[65, 15, 511], *neu_args, **neu_kwargs)

spikingjelly.activation_based.neuron_kernel.check_single_step_neuron_output_and_grad(device, single_step_neuron, shape=[65, 15, 511], *neu_args, **neu_kwargs)

class spikingjelly.activation_based.neuron_kernel.MultiStepQIFNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(hard_reset: bool, dtype: str)

static create_bptt_kernel(sg_cuda_code_fun, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, tau: float, v_threshold: float, v_reset: float, v_rest: float, v_c: float, a0: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq)

class spikingjelly.activation_based.neuron_kernel.MultiStepIzhikevichNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(hard_reset: bool, dtype: str)

static create_bptt_kernel(sg_cuda_code_fun, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, w_init: Tensor, tau: float, v_threshold: float, v_reset: float, v_rest: float, a: float, b: float, tau_w: float, v_c: float, a0: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq, grad_w_seq)

class spikingjelly.activation_based.neuron_kernel.MultiStepEIFNodePTT(*args, **kwargs)

基类：Function

static create_fptt_kernel(hard_reset: bool, dtype: str)

static create_bptt_kernel(sg_cuda_code_fun, hard_reset: bool, detach_reset: bool, dtype: str)

static forward(ctx, x_seq: Tensor, v_init: Tensor, tau: float, v_threshold: float, v_reset: float, v_rest: float, theta_rh: float, delta_T: float, detach_reset: bool, sg_cuda_code_fun)

static backward(ctx, grad_spike_seq, grad_v_seq)

spikingjelly.activation_based.neuron_kernel.save_cuda_codes(cu_file_path: str = './spikingjelly/activation_based/neuron_kernel_sample.cu')