tensor_utils.h File Reference

#include <ATen/core/tensor.h>
#include <ATen/core/tensor_shape.h>

Include dependency graph for tensor_utils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	container

Functions
__inline__ std::string	container::removeTrailingZeros (std::string str)
	Removes trailing zeros from a string.
template<typename T >
__inline__ int	container::_get_digit_places (const T *arr, int size, int &integer_count, int &fraction_count)
	Calculates the length of the longest integer and fractional part of an array of numbers.
template<typename T >
__inline__ int	container::_get_digit_places (const std::complex< T > *arr, int size, int &integer_count, int &fraction_count)
	Overloaded function to calculate the length of the longest integer and fractional part of an array of complex numbers.
template<typename T >
__inline__ void	container::_output_wrapper (std::ostream &os, const T data, const int &digit_width, const int &fraction_count)
	Output wrapper for a data value with given formatting parameters.
template<typename T >
__inline__ void	container::_output_wrapper (std::ostream &os, const std::complex< T > data, const int &digit_width, const int &fraction_count)
	Output wrapper for a complex data value with given formatting parameters.
template<>
__inline__ void	container::_output_wrapper (std::ostream &os, const int data, const int &digit_width, const int &fraction_count)
	Output wrapper for an integer data value with given formatting parameters.
template<typename T >
__inline__ void	container::_internal_output (std::ostream &os, const T *data, const TensorShape &shape, const int64_t &num_elements)
	Outputs tensor data to a given output stream.
template<typename T >
T	container::extract (const container::Tensor &tensor)