|
nnlib
GPU-accelerated, C/C++ neural network library.
|
Source file defining methods related to the Tensor class. More...
#include "tensor.h"#include "../gpu/allocation_gpu.cuh"#include "cache.h"#include "runtime.h"#include "tensor_operations_on_device.cuh"#include "tensor_operations_on_host.h"#include <exceptions/different_data_location_exception.h>#include <exceptions/size_mismatch_exception.h>#include <exceptions/unsupported_operation_exception.h>#include <functions.h>#include <memory>#include <queue>#include <string>#include <utils/location_verifiers.h>Functions | |
| bool | canBackPropagate (const Tensor &tensor) |
| std::string | tensorShapeToString (const Tensor &tensor) |
| Convert the shape of the tensor to a string. More... | |
| std::ostream & | operator<< (std::ostream &stream, const Tensor &tensor) |
| Enables the tensor to be printed using std::cout. More... | |
Source file defining methods related to the Tensor class.
| std::ostream& operator<< | ( | std::ostream & | stream, |
| const Tensor & | tensor | ||
| ) |
Enables the tensor to be printed using std::cout.
| stream | The stream to print the tensor to. |
| tensor | The tensor to print. |
| std::string tensorShapeToString | ( | const Tensor & | tensor | ) |
Convert the shape of the tensor to a string.
The shape is displayed in the format: [2, 2, 3].
| tensor | The tensor whose shape to show as a string. |