TensorVariable
小于 1 分钟
TensorVariable
TensorVariable Objects
class TensorVariable(Variable)
张量变量
张量变量是张量类型的变量载体, 提供了便可维度信息,形状、数据类型及填充器的管理功能。
getDim
def getDim(dim)
getDim(self: ocean.ocean.TensorVariable, dim: int) -> int
获取指定维度的维度大小
Parameter dim
: 维度
Returns:
维度大小
getTensor
def getTensor(*args, **kwargs)
getTensor(*args, **kwargs) Overloaded function.
- getTensor(self: ocean.ocean.TensorVariable) -> ocean.ocean.Tensor
获取到张量内容 Note: 返回张量不一定是存储张量本身,可能是其拷贝
Returns:
张量
- getTensor(self: ocean.ocean.TensorVariable, arg0: ocean.ocean.Tensor) -> ocean.ocean.Tensor
获取到张量内容,提供缓存区
Note: 返回张量不一定是存储张量本身,可能是其拷贝
Parameter buffer
: 如果buffer为空或者数据类型大小等不匹配,将重新创建新的buffer
Returns:
张量
dimension
张量维度数
filler
变量填充器
shape
张量形状
type
数据类型