bte.grad.collision_grad

Module Contents

Functions

load_collision_kernel([kernel_path, max_order, order, ...])
collision_f(Kernel, f)

Attributes

DATAPATH
MAX_DEG

bte.grad.collision_grad.DATAPATH = '/home/zyli/NRXXPKU/examples/test33/A/A_10_10.bin'

bte.grad.collision_grad.MAX_DEG = 10

bte.grad.collision_grad.load_collision_kernel(kernel_path=DATAPATH, max_order=MAX_DEG, order=None, dtype=torch.float32)

bte.grad.collision_grad.collision_f(Kernel, f)