BIGtensor-GPU
1.0
|
Compute binary operations for tensors. More...
Public Member Functions | |
void | TensorOP (char *in_tensor_A_path, OP op, char *in_tensor_B_path, char *out_tensor_path, const GPU_MODE _gpu_mode) |
Execution of basic binary operations with given two tensors. More... | |
void | EqualTo (char *in_tensor_A_path, char *in_tensor_B_path, char *out_tensor_path, const GPU_MODE _gpu_mode) |
Check if two tensor A and B are the same. More... | |
void | NModeProduct (char *in_tensor_path, char *in_sparse_matrix_path, int n_mode, char *out_tensor_path, const GPU_MODE _gpu_mode) |
N-mode product of a tensor and a sparse matrix. More... | |
Compute binary operations for tensors.
void Operation::EqualTo | ( | char * | in_tensor_A_path, |
char * | in_tensor_B_path, | ||
char * | out_tensor_path, | ||
const GPU_MODE | _gpu_mode | ||
) |
Check if two tensor A and B are the same.
in_tensor_A_path | input tensor A path |
in_tensor_B_path | input tensor B path |
out_tensor_path | output tensor path |
_gpu_mode | gpu mode for this computation (default: gpu_mode of a BIGtensor instance) |
None |
void Operation::NModeProduct | ( | char * | in_tensor_path, |
char * | in_sparse_matrix_path, | ||
int | n_mode, | ||
char * | out_tensor_path, | ||
const GPU_MODE | _gpu_mode | ||
) |
N-mode product of a tensor and a sparse matrix.
in_tensor_path | input tensor path |
in_sparse_matrix_path | input sparse matrix path |
n_mode | target mode |
out_tensor_path | output tensor path |
_gpu_mode | gpu mode for this computation (default: gpu_mode of a BIGtensor instance) |
None |
void Operation::TensorOP | ( | char * | in_tensor_A_path, |
OP | op, | ||
char * | in_tensor_B_path, | ||
char * | out_tensor_path, | ||
const GPU_MODE | _gpu_mode | ||
) |
Execution of basic binary operations with given two tensors.
in_tensor_A_path | input tensor A path |
op | operator (supporting: +, -, *, /) |
in_tensor_B_path | input tensor B path |
out_tensor_path | output tensor path |
_gpu_mode | gpu mode for this computation (default: gpu_mode of a BIGtensor instance) |
None |