PyTorch
Loading...
Searching...
No Matches
Namespaces | Functions
TensorShape.h File Reference
#include <ATen/core/Tensor.h>
#include <c10/util/irange.h>
#include <ATen/core/IListRef.h>

Go to the source code of this file.

Namespaces

namespace  at
 
namespace  at::native
 

Functions

at::Tensor at::native::clone_preserve_strides (const at::Tensor &self)
 
bool at::native::cat_should_skip_tensor (const Tensor &t)
 
void at::native::check_cat_shape_except_dim (const Tensor &first, const Tensor &second, int64_t dimension, int64_t index)
 
void at::native::check_cat_no_zero_dim (const MaterializedITensorListRef &tensors)
 
int64_t at::native::get_num_splits (const Tensor &self, int64_t split_size, int64_t dim)