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 () |
BCINLINE | Shape (bc::Dim< 1 > param) |
template<int X, class = std::enable_if_t<(X>=1> | |
BCINLINE | Shape (const Shape< X > &shape) |
BCINLINE | Shape (int length) |
BCINLINE size_t | operator[] (size_t i) const |
BCINLINE size_t | size () const |
BCINLINE size_t | rows () const |
BCINLINE size_t | cols () const |
BCINLINE size_t | dim (size_t i) const |
BCINLINE size_t | outer_dim () const |
BCINLINE size_t | leading_dim (size_t i=0) const |
BCINLINE const auto & | inner_shape () const |
BCINLINE bool | operator== (const Shape< 1 > &other) const |
template<class... Integers> | |
BCINLINE size_t | dims_to_index (size_t i, Integers... ints) const |
template<class... Integers> | |
BCINLINE size_t | dims_to_index (size_t i) const |
BCINLINE size_t | coefficientwise_dims_to_index (size_t i) const |
Static Public Attributes | |
static constexpr int | tensor_dim = 1 |
Friends | |
template<int > | |
struct | Shape |
using bc::Shape< 1 >::size_t = bc::size_t |
using bc::Shape< 1 >::value_type = size_t |
|
inline |
|
inline |
|
friend |
|
static |