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