TensorVariable

Neurocean小于 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.

  1. getTensor(self: ocean.ocean.TensorVariable) -> ocean.ocean.Tensor

获取到张量内容 Note: 返回张量不一定是存储张量本身,可能是其拷贝

Returns:

张量

  1. getTensor(self: ocean.ocean.TensorVariable, arg0: ocean.ocean.Tensor) -> ocean.ocean.Tensor

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

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

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

Returns:

张量

dimension

张量维度数

filler

变量填充器

shape

张量形状

type

数据类型

上次编辑于:
贡献者: damone