|
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 |