Tensor

Neurocean大约 2 分钟

Tensor

Tensor Objects

class Tensor()

张量

张量是一个存储载体,包括数据类型,数据布局,单元素字节大小,张量形状及数据等;

  • 提供了大量的内存分配及数据引用张量的构建方法;
  • 提供了 create_like, fill, empty, total, 方括号引用的操作的支持接口;

__getitem__

def __getitem__(i)

getitem(self: ocean.ocean.Tensor, i: int) -> object

根据索引访问元素

Parameter i: 索引

Returns:

数值

__init__

def __init__(*args, **kwargs)

init(*args, **kwargs) Overloaded function.

  1. init(self: ocean.ocean.Tensor) -> None

空张量

  1. init(self: ocean.ocean.Tensor, tensor: ocean.ocean.Tensor) -> None

拷贝张量

Parameter m: 待拷贝张量

  1. init(self: ocean.ocean.Tensor, dim0: int, *, type: ocean.ocean.DataType = <DataType.Float: 3>, layout: ocean.ocean.Layout = <Layout.TZYX: 3>) -> None

构建一维向量

Parameter dim0: 一维数据长度

Parameter type: 数据类型

Parameter layout: 数据布局

  1. init(self: ocean.ocean.Tensor, dim0: int, dim1: int, *, type: ocean.ocean.DataType = <DataType.Float: 3>, layout: ocean.ocean.Layout = <Layout.TZYX: 3>) -> None

构建二维矩阵

Parameter dim0: 维度0

Parameter dim1: 维度1

Parameter type: 数据类型

Parameter layout: 数据布局

  1. init(self: ocean.ocean.Tensor, dim0: int, dim1: int, dim2: int, *, type: ocean.ocean.DataType = <DataType.Float: 3>, layout: ocean.ocean.Layout = <Layout.TZYX: 3>) -> None

构建三维张量

Parameter dim0: 维度0

Parameter dim1: 维度1

Parameter dim2: 维度2

Parameter type: 数据类型

Parameter layout: 数据布局

  1. init(self: ocean.ocean.Tensor, dim0: int, dim1: int, dim2: int, dim3: int, *, type: ocean.ocean.DataType = <DataType.Float: 3>, layout: ocean.ocean.Layout = <Layout.TZYX: 3>) -> None

构建四维张量

Parameter dim0: 维度0

Parameter dim1: 维度1

Parameter dim2: 维度2

Parameter dim3: 维度3

Parameter type: 数据类型

Parameter layout: 数据布局

  1. init(self: ocean.ocean.Tensor, shape: tuple, *, type: ocean.ocean.DataType = <DataType.Float: 3>, layout: ocean.ocean.Layout = <Layout.TZYX: 3>) -> None

从形状构建张量

Parameter shape: 形状

Parameter type: 数据类型

Parameter layout: 数据布局

  1. init(self: ocean.ocean.Tensor, array: buffer) -> None

从数据构建张量

Parameter array: 维度 0

__len__

def __len__()

len(self: ocean.ocean.Tensor) -> int

数据元素总数

__setitem__

def __setitem__(i, v)

setitem(self: ocean.ocean.Tensor, i: int, v: object) -> None

根据索引设置元素

Parameter i: 索引

Parameter v: 数值

clone

def clone()

clone(self: ocean.ocean.Tensor) -> ocean.ocean.Tensor

深度拷贝

Returns:

张量

create_like

def create_like(tensor)

create_like(self: ocean.ocean.Tensor, tensor: ocean.ocean.Tensor) -> None

创建相似形状张量,不拷贝数据

Parameter like: 被相似张量

fill

def fill(v)

fill(self: ocean.ocean.Tensor, v: object) -> None

以指定数值填充张量

Parameter value:

reshape

def reshape(arg0)

reshape(self: ocean.ocean.Tensor, arg0: tuple) -> ocean.ocean.Tensor

重新调整形状并返回调整后的张量

Parameter shape: 新的 shape

Returns:

张量

shape

def shape()

shape(self: ocean.ocean.Tensor) -> tuple

张量形状

elemSize

元素字节大小

empty

是否为空

layout

数据布局

total

数据元素总数

type

数据类型

上次编辑于:
贡献者: damone