ocean::TensorVariable

张量变量 More...

#include <Variable.hpp>

Inherits from ocean::Variable, std::enable_shared_from_this< Variable >

Public Functions

	Name
	TensorVariable(PIVariable variable) 构建张量变量
int	getDimension() const 获取张量维度数
uint64_t	getDim(int dim) 获取指定维度的维度大小
const TensorShape &	getShape() const 获取张量形状
DataType	getType() const 获取数据类型
const Filler	getFiller() const 获取变量填充器
Tensor	getTensor() 获取到张量内容
Tensor	getTensor(Tensor & buffer) 获取到张量内容，提供缓存区

Additional inherited members

Public Functions inherited from ocean::Variable

	Name
	Variable(PIVariable variable) 构建变量
virtual	~Variable() 析构变量
const std::string &	getName() const 获取变量名称
const std::string &	getDescription() const 获取变量描述
void	setDescription(const std::string & description) 设置变量描述
bool	isShared() const 是否共享
bool	isPersistent() const 是否持久化，只对张量有效
Layout	getLayout() const 获取变量布局
PIVariable	getInternalVariable() 获取内部变量指针

Protected Attributes inherited from ocean::Variable

	Name
PIVariable	internalVariable 内部变量指针

Detailed Description

class ocean::TensorVariable;

张量变量

张量变量是张量类型的变量载体， 提供了便可维度信息，形状、数据类型及填充器的管理功能。

Public Functions Documentation

function TensorVariable

explicit TensorVariable(
    PIVariable variable
)

构建张量变量

Parameters:

• variable 张量变量

function getDimension

int getDimension() const

获取张量维度数

Return: 维度数

function getDim

uint64_t getDim(
    int dim
)

获取指定维度的维度大小

Parameters:

• dim 维度

Return: 维度大小

function getShape

const TensorShape & getShape() const

获取张量形状

Return: 张量形状

function getType

DataType getType() const

获取数据类型

Return: 数据类型

function getFiller

const Filler getFiller() const

获取变量填充器

Return:

function getTensor

Tensor getTensor()

获取到张量内容

Return: 张量

Note: 返回张量不一定是存储张量本身，可能是其拷贝

function getTensor

Tensor getTensor(
    Tensor & buffer
)

获取到张量内容，提供缓存区

Parameters:

• buffer 如果buffer为空或者数据类型大小等不匹配，将重新创建新的buffer

Return: 张量

Note: 返回张量不一定是存储张量本身，可能是其拷贝

---

Updated on 2022-08-19 at 17:14:26 +0800