BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
#include <shape.h>
Public Types | |
using | size_t = bc::size_t |
using | value_type = size_t |
Public Member Functions | |
BCINLINE | Shape () |
template<class... Args> | |
BCINLINE | Shape (const Args &...) |
BCINLINE Dim< 0 > | inner_shape () const |
BCINLINE size_t | operator[] (size_t i) |
BCINLINE size_t | size () const |
BCINLINE size_t | rows () const |
BCINLINE size_t | cols () const |
BCINLINE size_t | dim (int i) const |
BCINLINE size_t | outer_dim () const |
BCINLINE size_t | leading_dim (int i=0) const |
BCINLINE bool | operator== (const Shape &other) const |
BCINLINE size_t | coefficientwise_dims_to_index (size_t i) const |
template<class... Integers> | |
BCINLINE size_t | dims_to_index (Integers... ints) const |
Static Public Attributes | |
static constexpr int | tensor_dim = 0 |
Friends | |
template<int > | |
struct | Shape |
using bc::Shape< 0 >::size_t = bc::size_t |
using bc::Shape< 0 >::value_type = size_t |
|
inline |
|
friend |
|
static |