|
BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
#include "layer_base.h"

Go to the source code of this file.
Classes | |
| struct | bc::nn::Flatten< SystemTag, ValueType, InputTensorDimension, InputDescriptor, OutputDescriptor > |
Namespaces | |
| bc | |
| The Evaluator determines if an expression needs to be greedily optimized. | |
| bc::nn | |
Functions | |
| template<class ValueType , class SystemTag , int X> | |
| auto | bc::nn::flatten (SystemTag system_tag, Dim< X > shape) |
| template<class SystemTag , int X> | |
| auto | bc::nn::flatten (SystemTag system_tag, Dim< X > shape) |
| template<int X> | |
| auto | bc::nn::flatten (Dim< X > shape) |
1.8.13