surkit.backend package

Submodules

surkit.backend.backend module

surkit.backend.backend.arccos(tensor)

surkit.backend.backend.arccosh(tensor)

surkit.backend.backend.arcsin(tensor)

surkit.backend.backend.arcsinh(tensor)

surkit.backend.backend.arctan(tensor)

surkit.backend.backend.arctanh(tensor)

surkit.backend.backend.cat(tensor_list, dim)

surkit.backend.backend.cos(tensor)

surkit.backend.backend.cosh(tensor)

surkit.backend.backend.exp(tensor)

surkit.backend.backend.forward(model, x)

surkit.backend.backend.grad()

surkit.backend.backend.is_tensor(obj)

surkit.backend.backend.log(tensor)

surkit.backend.backend.np_to_tensor(array)

surkit.backend.backend.power(tensor, exponent)

surkit.backend.backend.save()

surkit.backend.backend.sin(tensor)

surkit.backend.backend.sinh(tensor)

surkit.backend.backend.sqrt(tensor)

surkit.backend.backend.squeeze(tensor, dim)

surkit.backend.backend.tan(tensor)

surkit.backend.backend.tanh(tensor)

surkit.backend.backend.unsqueeze(tensor, dim)

surkit.backend.backend.zeros(shape, dtype)

surkit.backend.backend.zeros_like(tensor)

surkit.backend.jax_bkd module

surkit.backend.jax_bkd.arccos(tensor)

surkit.backend.jax_bkd.arccosh(tensor)

surkit.backend.jax_bkd.arcsin(tensor)

surkit.backend.jax_bkd.arcsinh(tensor)

surkit.backend.jax_bkd.arctan(tensor)

surkit.backend.jax_bkd.arctanh(tensor)

surkit.backend.jax_bkd.cat(tensor_list, dim=0)

surkit.backend.jax_bkd.cos(tensor)

surkit.backend.jax_bkd.cosh(tensor)

surkit.backend.jax_bkd.exp(tensor)

surkit.backend.jax_bkd.forward(model, x, params=None)

surkit.backend.jax_bkd.grad(state, x, ind_x, ind_y)

Calculate dy/dx.

Parameters:

• y (tensor) –

• x (tensor) –

Returns:

dy/dx

surkit.backend.jax_bkd.is_tensor(obj)

surkit.backend.jax_bkd.load(state, path)

surkit.backend.jax_bkd.log(tensor)

surkit.backend.jax_bkd.log10(tensor)

surkit.backend.jax_bkd.log2(tensor)

surkit.backend.jax_bkd.np_to_tensor(array)

surkit.backend.jax_bkd.power(tensor, exponent)

surkit.backend.jax_bkd.save(state, path)

surkit.backend.jax_bkd.sin(tensor)

surkit.backend.jax_bkd.sinh(tensor)

surkit.backend.jax_bkd.sqrt(tensor)

surkit.backend.jax_bkd.squeeze(tensor, dim)

surkit.backend.jax_bkd.tan(tensor)

surkit.backend.jax_bkd.tanh(tensor)

surkit.backend.jax_bkd.unsqueeze(tensor, dim)

surkit.backend.jax_bkd.zeros(shape, dtype=None)

surkit.backend.jax_bkd.zeros_like(tensor)

surkit.backend.oneflow_bkd module

surkit.backend.pytorch_bkd module

surkit.backend.pytorch_bkd.arccos(tensor)

surkit.backend.pytorch_bkd.arccosh(tensor)

surkit.backend.pytorch_bkd.arcsin(tensor)

surkit.backend.pytorch_bkd.arcsinh(tensor)

surkit.backend.pytorch_bkd.arctan(tensor)

surkit.backend.pytorch_bkd.arctanh(tensor)

surkit.backend.pytorch_bkd.cat(tensor_list, dim=0)

surkit.backend.pytorch_bkd.cos(tensor)

surkit.backend.pytorch_bkd.cosh(tensor)

surkit.backend.pytorch_bkd.exp(tensor)

surkit.backend.pytorch_bkd.forward(model, x)

surkit.backend.pytorch_bkd.grad(y, x)

Calculate dy/dx.

Parameters:

• y (tensor) –

• x (tensor) –

Returns:

dy/dx

surkit.backend.pytorch_bkd.is_tensor(obj)

surkit.backend.pytorch_bkd.load(path)

surkit.backend.pytorch_bkd.log(tensor)

surkit.backend.pytorch_bkd.log10(tensor)

surkit.backend.pytorch_bkd.log2(tensor)

surkit.backend.pytorch_bkd.np_to_tensor(array, requires_grad=False)

surkit.backend.pytorch_bkd.power(tensor, exponent)

surkit.backend.pytorch_bkd.save(model, path)

surkit.backend.pytorch_bkd.sin(tensor)

surkit.backend.pytorch_bkd.sinh(tensor)

surkit.backend.pytorch_bkd.sqrt(tensor)

surkit.backend.pytorch_bkd.squeeze(tensor, dim)

surkit.backend.pytorch_bkd.tan(tensor)

surkit.backend.pytorch_bkd.tanh(tensor)

surkit.backend.pytorch_bkd.unsqueeze(tensor, dim)

surkit.backend.pytorch_bkd.zeros(shape, dtype=None)

surkit.backend.pytorch_bkd.zeros_like(tensor)

Module contents