taichi._kernels

Module Contents

Functions

fill_tensor(tensor: template(), val: template())
fill_ndarray(ndarray: any_arr(), val: template())
fill_ndarray_matrix(ndarray: any_arr(), val: template())
tensor_to_ext_arr(tensor: template(), arr: ext_arr())
ndarray_to_ext_arr(ndarray: any_arr(), arr: ext_arr())
ndarray_matrix_to_ext_arr(ndarray: any_arr(), arr: ext_arr(), as_vector: template())
vector_to_fast_image(img: template(), out: ext_arr())
tensor_to_image(tensor: template(), arr: ext_arr())
vector_to_image(mat: template(), arr: ext_arr())
tensor_to_tensor(tensor: template(), other: template())
ext_arr_to_tensor(arr: ext_arr(), tensor: template())
ndarray_to_ndarray(ndarray: any_arr(), other: any_arr())
ext_arr_to_ndarray(arr: ext_arr(), ndarray: any_arr())
ext_arr_to_ndarray_matrix(arr: ext_arr(), ndarray: any_arr(), as_vector: template())
matrix_to_ext_arr(mat: template(), arr: ext_arr(), as_vector: template())
ext_arr_to_matrix(arr: ext_arr(), mat: template(), as_vector: template())
clear_gradients(_vars: template())
clear_loss(l: template())
fill_matrix(mat: template(), vals: template())
snode_deactivate(b: template())
snode_deactivate_dynamic(b: template())

taichi._kernels.fill_tensor(tensor: template(), val: template())

taichi._kernels.fill_ndarray(ndarray: any_arr(), val: template())

taichi._kernels.fill_ndarray_matrix(ndarray: any_arr(), val: template())

taichi._kernels.tensor_to_ext_arr(tensor: template(), arr: ext_arr())

taichi._kernels.ndarray_to_ext_arr(ndarray: any_arr(), arr: ext_arr())

taichi._kernels.ndarray_matrix_to_ext_arr(ndarray: any_arr(), arr: ext_arr(), as_vector: template())

taichi._kernels.vector_to_fast_image(img: template(), out: ext_arr())

taichi._kernels.tensor_to_image(tensor: template(), arr: ext_arr())

taichi._kernels.vector_to_image(mat: template(), arr: ext_arr())

taichi._kernels.tensor_to_tensor(tensor: template(), other: template())

taichi._kernels.ext_arr_to_tensor(arr: ext_arr(), tensor: template())

taichi._kernels.ndarray_to_ndarray(ndarray: any_arr(), other: any_arr())

taichi._kernels.ext_arr_to_ndarray(arr: ext_arr(), ndarray: any_arr())

taichi._kernels.ext_arr_to_ndarray_matrix(arr: ext_arr(), ndarray: any_arr(), as_vector: template())

taichi._kernels.matrix_to_ext_arr(mat: template(), arr: ext_arr(), as_vector: template())

taichi._kernels.ext_arr_to_matrix(arr: ext_arr(), mat: template(), as_vector: template())

taichi._kernels.clear_gradients(_vars: template())

taichi._kernels.clear_loss(l: template())

taichi._kernels.fill_matrix(mat: template(), vals: template())

taichi._kernels.snode_deactivate(b: template())

taichi._kernels.snode_deactivate_dynamic(b: template())