eve::tensor::TF类 参考

TF2-like namespace module. Script: tf <- eve.TF(); Default eager; tf.func() traces a graph for compile/run. 更多...

#include <TF.h>

类 eve::tensor::TF 继承关系图:

Public 成员函数
	Module_REG (TF)
	TF ()
	Tf.
	~TF () override=default
	Tf.
Func *	func ()
	Func.
void	pushTrace (Func *f)
	Pushes trace.
void	popTrace (Func *f)
	Pops trace.
Func *	tracing () const
	Tracing.
Tensor *	zeros1 (int d0)
	Zeros 1.
Tensor *	zeros2 (int d0, int d1)
	Zeros 2.
Tensor *	zeros3 (int d0, int d1, int d2)
	Zeros 3.
Tensor *	zeros4 (int d0, int d1, int d2, int d3)
	Zeros 4.
Tensor *	zeros5 (int d0, int d1, int d2, int d3, int d4)
	Zeros 5.
Tensor *	zeros6 (int d0, int d1, int d2, int d3, int d4, int d5)
	Zeros 6.
Tensor *	ones1 (int d0)
	Ones 1.
Tensor *	ones2 (int d0, int d1)
	Ones 2.
Tensor *	ones3 (int d0, int d1, int d2)
	Ones 3.
Tensor *	ones4 (int d0, int d1, int d2, int d3)
	Ones 4.
Tensor *	ones5 (int d0, int d1, int d2, int d3, int d4)
	Ones 5.
Tensor *	ones6 (int d0, int d1, int d2, int d3, int d4, int d5)
	Ones 6.
Tensor *	fill1 (int d0, float value)
	Fill 1.
Tensor *	fill2 (int d0, int d1, float value)
	Fill 2.
Tensor *	fill3 (int d0, int d1, int d2, float value)
	Fill 3.
Tensor *	fill4 (int d0, int d1, int d2, int d3, float value)
	Fill 4.
Tensor *	constantScalar (float value)
	Constant scalar.
Tensor *	arange (int n)
	Arange.
Tensor *	linspace (float start, float end, int n)
	Linspace.
Tensor *	eye (int n)
	Eye.
Tensor *	randomUniform1 (int d0)
	Random uniform 1.
Tensor *	randomUniform2 (int d0, int d1)
	Random uniform 2.
Tensor *	randomUniform3 (int d0, int d1, int d2)
	Random uniform 3.
Tensor *	randomUniform4 (int d0, int d1, int d2, int d3)
	Random uniform 4.
Tensor *	randomNormal1 (int d0)
	Random normal 1.
Tensor *	randomNormal2 (int d0, int d1)
	Random normal 2.
Tensor *	randomNormal3 (int d0, int d1, int d2)
	Random normal 3.
Tensor *	randomNormal4 (int d0, int d1, int d2, int d3)
	Random normal 4.
Tensor *	rand1 (int d0)
	Rand 1.
Tensor *	rand2 (int d0, int d1)
	Rand 2.
Tensor *	rand3 (int d0, int d1, int d2)
	Rand 3.
Tensor *	rand4 (int d0, int d1, int d2, int d3)
	Rand 4.
Tensor *	randn1 (int d0)
	Randn 1.
Tensor *	randn2 (int d0, int d1)
	Randn 2.
Tensor *	randn3 (int d0, int d1, int d2)
	Randn 3.
Tensor *	randn4 (int d0, int d1, int d2, int d3)
	Randn 4.
void	setRandomSeed (uint32_t seed)
	Sets the random seed.
uint32_t	getRandomSeed () const
	Returns the random seed.
Tensor *	add (Tensor *a, Tensor *b)
	Adds add.
Tensor *	sub (Tensor *a, Tensor *b)
	Sub.
Tensor *	multiply (Tensor *a, Tensor *b)
	Multiply.
Tensor *	div (Tensor *a, Tensor *b)
	Div.
Tensor *	addScalar (Tensor *a, float s)
	Adds scalar.
Tensor *	subScalar (Tensor *a, float s)
	Sub scalar.
Tensor *	mulScalar (Tensor *a, float s)
	Mul scalar.
Tensor *	divScalar (Tensor *a, float s)
	Div scalar.
Tensor *	neg (Tensor *a)
	Neg.
Tensor *	abs (Tensor *a)
	Abs.
Tensor *	sqrt (Tensor *a)
	Sqrt.
Tensor *	exp (Tensor *a)
	Exp.
Tensor *	log (Tensor *a)
	Log.
Tensor *	sin (Tensor *a)
	Sin.
Tensor *	cos (Tensor *a)
	Cos.
Tensor *	tanh (Tensor *a)
	Tanh.
Tensor *	relu (Tensor *a)
	Relu.
Tensor *	sigmoid (Tensor *a)
	Sigmoid.
Tensor *	gelu (Tensor *a)
	Gelu.
Tensor *	silu (Tensor *a)
	Silu.
Tensor *	powScalar (Tensor *a, float exp)
	Pow scalar.
Tensor *	clamp (Tensor *a, float lo, float hi)
	Clamp.
Tensor *	maximumScalar (Tensor *a, float s)
	Maximum scalar.
Tensor *	minimumScalar (Tensor *a, float s)
	Minimum scalar.
Tensor *	matmul (Tensor *a, Tensor *b)
	Matmul.
Tensor *	transpose (Tensor *a)
	Transpose.
Tensor *	permute2 (Tensor *a, int a0, int a1)
	Permute 2.
Tensor *	permute3 (Tensor *a, int a0, int a1, int a2)
	Permute 3.
Tensor *	permute4 (Tensor *a, int a0, int a1, int a2, int a3)
	Permute 4.
Tensor *	permute5 (Tensor *a, int a0, int a1, int a2, int a3, int a4)
	Permute 5.
Tensor *	permute6 (Tensor *a, int a0, int a1, int a2, int a3, int a4, int a5)
	Permute 6.
Tensor *	reshape1 (Tensor *a, int d0)
	Reshape 1.
Tensor *	reshape2 (Tensor *a, int d0, int d1)
	Reshape 2.
Tensor *	reshape3 (Tensor *a, int d0, int d1, int d2)
	Reshape 3.
Tensor *	reshape4 (Tensor *a, int d0, int d1, int d2, int d3)
	Reshape 4.
Tensor *	reshape5 (Tensor *a, int d0, int d1, int d2, int d3, int d4)
	Reshape 5.
Tensor *	reshape6 (Tensor *a, int d0, int d1, int d2, int d3, int d4, int d5)
	Reshape 6.
Tensor *	flatten (Tensor *a)
	Flatten.
Tensor *	where (Tensor *cond, Tensor *a, Tensor *b)
	Where.
Tensor *	concatN (Tensor *const *ins, int n, int axis)
	Concat n.
Tensor *	softmax (Tensor *a, int axis)
	Softmax.
Tensor *	logSoftmax (Tensor *a, int axis)
	Log softmax.
Tensor *	layernorm (Tensor *a, float eps)
	Layernorm.
Tensor *	layernormWB (Tensor *a, Tensor *scale, Tensor *bias, float eps)
	Layernorm wb.
Tensor *	rmsnorm (Tensor *a, float eps)
	Rmsnorm.
Tensor *	rmsnormW (Tensor *a, Tensor *scale, float eps)
	Rmsnorm w.
Tensor *	conv1d (Tensor *x, Tensor *w, int stride, int pad)
	Conv 1 d.
Tensor *	conv1dBias (Tensor *x, Tensor *w, Tensor *bias, int stride, int pad)
	Conv 1 d bias.
Tensor *	conv2d (Tensor *x, Tensor *w, int stride, int pad)
	Conv 2 d.
Tensor *	conv2dBias (Tensor *x, Tensor *w, Tensor *bias, int stride, int pad)
	Conv 2 d bias.
Tensor *	maxpool2d (Tensor *x, int ksize, int stride, int pad)
	Maxpool 2 d.
Tensor *	avgpool2d (Tensor *x, int ksize, int stride, int pad)
	Avgpool 2 d.
Tensor *	embedding (Tensor *table, Tensor *indices)
	Embedding.
Tensor *	concat2 (Tensor *a, Tensor *b, int axis)
	Concat 2.
Tensor *	concat3 (Tensor *a, Tensor *b, Tensor *c, int axis)
	Concat 3.
Tensor *	concat4 (Tensor *a, Tensor *b, Tensor *c, Tensor *d, int axis)
	Concat 4.
Tensor *	slice (Tensor *a, int axis, int begin, int end)
	Slice.
Tensor *	sumAxis (Tensor *a, int axis, int keepDims)
	Sum axis.
Tensor *	meanAxis (Tensor *a, int axis, int keepDims)
	Mean axis.
Tensor *	minAxis (Tensor *a, int axis, int keepDims)
	Min axis.
Tensor *	maxAxis (Tensor *a, int axis, int keepDims)
	Max axis.
Tensor *	argmax (Tensor *a, int axis, int keepDims)
	Argmax.
Tensor *	cast (Tensor *a, const std::string &dtype)
	Cast.
Tensor *	sdpa (Tensor *q, Tensor *k, Tensor *v, float scale)
	Sdpa.
Tensor *	sdpaMasked (Tensor *q, Tensor *k, Tensor *v, Tensor *mask, float scale)
	Sdpa masked.
Tensor *	resize2d (Tensor *a, int outW, int outH, int mode)
	Resize 2 d.
Tensor *	quantizeWeight (Tensor *a, const std::string &dtype, int group=0)
	Quantize weight.
float	reduceSum (Tensor *a)
	Reduce sum.
float	reduceMean (Tensor *a)
	Reduce mean.
float	reduceMin (Tensor *a)
	Reduce min.
float	reduceMax (Tensor *a)
	Reduce max.
Public 成员函数 继承自 eve::Module
virtual	~Module ()
	Module.
virtual std::string	getName () const =0
	Returns the name.

详细描述

TF2-like namespace module. Script: tf <- eve.TF(); Default eager; tf.func() traces a graph for compile/run.

在文件 TF.h 第 20 行定义.

构造及析构函数说明

TF()

eve::tensor::TF::TF

(

)

Tf.

在文件 TF.cpp 第 52 行定义.

~TF()

eve::tensor::TF::~TF ( )

overridedefault

Tf.

成员函数说明

abs()

Tensor * eve::tensor::TF::abs

(

Tensor *

a

)

Abs.

add()

Tensor * eve::tensor::TF::add	(	Tensor *	a,
		Tensor *	b
	)

Adds add.

addScalar()

Tensor * eve::tensor::TF::addScalar	(	Tensor *	a,
		float	s
	)

Adds scalar.

在文件 TF.cpp 第 307 行定义.

引用了 a, eve::tensor::AddScalar, f, s , 以及 tracing().

arange()

Tensor * eve::tensor::TF::arange

(

int

n

)

Arange.

在文件 TF.cpp 第 176 行定义.

引用了 n, t , 以及 tracing().

argmax()

Tensor * eve::tensor::TF::argmax	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

Argmax.

在文件 TF.cpp 第 733 行定义.

引用了 a, eve::tensor::kernels::argmax(), f, eve::tensor::Int32, eve::tensor::Tensor::kMaxRank , 以及 tracing().

avgpool2d()

Tensor * eve::tensor::TF::avgpool2d	(	Tensor *	x,
		int	ksize,
		int	stride,
		int	pad
	)

Avgpool 2 d.

在文件 TF.cpp 第 585 行定义.

引用了 eve::tensor::kernels::avgpool2d(), eve::tensor::AvgPool2d, f, pad, tracing() , 以及 x.

cast()

Tensor * eve::tensor::TF::cast	(	Tensor *	a,
		const std::string &	dtype
	)

Cast.

在文件 TF.cpp 第 748 行定义.

引用了 a, f, eve::tensor::Float32, eve::tensor::Tensor::kMaxRank, eve::tensor::parseDType() , 以及 tracing().

clamp()

Tensor * eve::tensor::TF::clamp	(	Tensor *	a,
		float	lo,
		float	hi
	)

Clamp.

在文件 TF.cpp 第 332 行定义.

引用了 a, eve::tensor::Clamp, f , 以及 tracing().

concat2()

Tensor * eve::tensor::TF::concat2	(	Tensor *	a,
		Tensor *	b,
		int	axis
	)

Concat 2.

在文件 TF.cpp 第 633 行定义.

引用了 a, b , 以及 concatN().

concat3()

Tensor * eve::tensor::TF::concat3	(	Tensor *	a,
		Tensor *	b,
		Tensor *	c,
		int	axis
	)

Concat 3.

在文件 TF.cpp 第 638 行定义.

引用了 a, b, c , 以及 concatN().

concat4()

Tensor * eve::tensor::TF::concat4	(	Tensor *	a,
		Tensor *	b,
		Tensor *	c,
		Tensor *	d,
		int	axis
	)

Concat 4.

在文件 TF.cpp 第 643 行定义.

引用了 a, b, c, concatN() , 以及 d.

concatN()

Tensor * eve::tensor::TF::concatN	(	Tensor *const *	ins,
		int	n,
		int	axis
	)

Concat n.

在文件 TF.cpp 第 684 行定义.

引用了 f, n , 以及 tracing().

被这些函数引用 concat2(), concat3() , 以及 concat4().

constantScalar()

Tensor * eve::tensor::TF::constantScalar

(

float

value

)

Constant scalar.

在文件 TF.cpp 第 171 行定义.

引用了 d , 以及 value.

conv1d()

Tensor * eve::tensor::TF::conv1d	(	Tensor *	x,
		Tensor *	w,
		int	stride,
		int	pad
	)

Conv 1 d.

在文件 TF.cpp 第 534 行定义.

引用了 conv1dBias(), pad, w , 以及 x.

conv1dBias()

Tensor * eve::tensor::TF::conv1dBias	(	Tensor *	x,
		Tensor *	w,
		Tensor *	bias,
		int	stride,
		int	pad
	)

Conv 1 d bias.

在文件 TF.cpp 第 538 行定义.

引用了 bias, eve::tensor::kernels::conv1d(), f, pad, tracing(), w , 以及 x.

被这些函数引用 conv1d().

conv2d()

Tensor * eve::tensor::TF::conv2d	(	Tensor *	x,
		Tensor *	w,
		int	stride,
		int	pad
	)

Conv 2 d.

在文件 TF.cpp 第 553 行定义.

引用了 conv2dBias(), pad, w , 以及 x.

conv2dBias()

Tensor * eve::tensor::TF::conv2dBias	(	Tensor *	x,
		Tensor *	w,
		Tensor *	bias,
		int	stride,
		int	pad
	)

Conv 2 d bias.

在文件 TF.cpp 第 557 行定义.

引用了 bias, eve::tensor::kernels::conv2d(), f, pad, tracing(), w , 以及 x.

被这些函数引用 conv2d().

cos()

Tensor * eve::tensor::TF::cos

(

Tensor *

a

)

Cos.

div()

Tensor * eve::tensor::TF::div	(	Tensor *	a,
		Tensor *	b
	)

Div.

divScalar()

Tensor * eve::tensor::TF::divScalar	(	Tensor *	a,
		float	s
	)

Div scalar.

在文件 TF.cpp 第 322 行定义.

引用了 a, eve::tensor::DivScalar, f, s , 以及 tracing().

embedding()

Tensor * eve::tensor::TF::embedding	(	Tensor *	table,
		Tensor *	indices
	)

Embedding.

在文件 TF.cpp 第 597 行定义.

引用了 eve::tensor::kernels::embedding(), f, indices, eve::tensor::Tensor::kMaxRank , 以及 tracing().

exp()

Tensor * eve::tensor::TF::exp

(

Tensor *

a

)

Exp.

被这些函数引用 powScalar().

eye()

Tensor * eve::tensor::TF::eye

(

int

n

)

Eye.

在文件 TF.cpp 第 197 行定义.

引用了 f, eve::tensor::Tensor::makeSymbolic(), n, t , 以及 tracing().

fill1()

Tensor * eve::tensor::TF::fill1	(	int	d0,
		float	value
	)

Fill 1.

在文件 TF.cpp 第 154 行定义.

引用了 d , 以及 value.

fill2()

Tensor * eve::tensor::TF::fill2	(	int	d0,
		int	d1,
		float	value
	)

Fill 2.

在文件 TF.cpp 第 158 行定义.

引用了 d , 以及 value.

fill3()

Tensor * eve::tensor::TF::fill3	(	int	d0,
		int	d1,
		int	d2,
		float	value
	)

Fill 3.

在文件 TF.cpp 第 162 行定义.

引用了 d , 以及 value.

fill4()

Tensor * eve::tensor::TF::fill4	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		float	value
	)

Fill 4.

在文件 TF.cpp 第 166 行定义.

引用了 d , 以及 value.

flatten()

Tensor * eve::tensor::TF::flatten

(

Tensor *

a

)

Flatten.

在文件 TF.cpp 第 428 行定义.

引用了 a, d, f , 以及 tracing().

func()

Func * eve::tensor::TF::func

(

)

Func.

在文件 TF.cpp 第 95 行定义.

gelu()

Tensor * eve::tensor::TF::gelu

(

Tensor *

a

)

Gelu.

getRandomSeed()

uint32_t eve::tensor::TF::getRandomSeed

(

)

const

Returns the random seed.

在文件 TF.cpp 第 71 行定义.

layernorm()

Tensor * eve::tensor::TF::layernorm	(	Tensor *	a,
		float	eps
	)

Layernorm.

在文件 TF.cpp 第 479 行定义.

引用了 a, cols, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::layernorm(), rows , 以及 tracing().

layernormWB()

Tensor * eve::tensor::TF::layernormWB	(	Tensor *	a,
		Tensor *	scale,
		Tensor *	bias,
		float	eps
	)

Layernorm wb.

在文件 TF.cpp 第 492 行定义.

引用了 a, bias, cols, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::layernorm(), rows, scale , 以及 tracing().

linspace()

Tensor * eve::tensor::TF::linspace	(	float	start,
		float	end,
		int	n
	)

Linspace.

在文件 TF.cpp 第 184 行定义.

引用了 end, n, start, step, t , 以及 tracing().

log()

Tensor * eve::tensor::TF::log

(

Tensor *

a

)

Log.

logSoftmax()

Tensor * eve::tensor::TF::logSoftmax	(	Tensor *	a,
		int	axis
	)

Log softmax.

在文件 TF.cpp 第 467 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::softmax() , 以及 tracing().

matmul()

Tensor * eve::tensor::TF::matmul	(	Tensor *	a,
		Tensor *	b
	)

Matmul.

在文件 TF.cpp 第 348 行定义.

引用了 a, b, f , 以及 tracing().

maxAxis()

Tensor * eve::tensor::TF::maxAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

Max axis.

maximumScalar()

Tensor * eve::tensor::TF::maximumScalar	(	Tensor *	a,
		float	s
	)

Maximum scalar.

在文件 TF.cpp 第 337 行定义.

引用了 a, f, eve::tensor::MaximumScalar, s , 以及 tracing().

maxpool2d()

Tensor * eve::tensor::TF::maxpool2d	(	Tensor *	x,
		int	ksize,
		int	stride,
		int	pad
	)

Maxpool 2 d.

在文件 TF.cpp 第 573 行定义.

引用了 f, eve::tensor::kernels::maxpool2d(), eve::tensor::MaxPool2d, pad, tracing() , 以及 x.

meanAxis()

Tensor * eve::tensor::TF::meanAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

Mean axis.

minAxis()

Tensor * eve::tensor::TF::minAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

Min axis.

minimumScalar()

Tensor * eve::tensor::TF::minimumScalar	(	Tensor *	a,
		float	s
	)

Minimum scalar.

在文件 TF.cpp 第 342 行定义.

引用了 a, f, eve::tensor::MinimumScalar, s , 以及 tracing().

Module_REG()

eve::tensor::TF::Module_REG

(

TF

)

mulScalar()

Tensor * eve::tensor::TF::mulScalar	(	Tensor *	a,
		float	s
	)

Mul scalar.

在文件 TF.cpp 第 317 行定义.

引用了 a, f, eve::tensor::MulScalar, s , 以及 tracing().

multiply()

Tensor * eve::tensor::TF::multiply	(	Tensor *	a,
		Tensor *	b
	)

Multiply.

neg()

Tensor * eve::tensor::TF::neg

(

Tensor *

a

)

Neg.

ones1()

Tensor * eve::tensor::TF::ones1

(

int

d0

)

Ones 1.

在文件 TF.cpp 第 129 行定义.

引用了 d.

ones2()

Tensor * eve::tensor::TF::ones2	(	int	d0,
		int	d1
	)

Ones 2.

在文件 TF.cpp 第 133 行定义.

引用了 d.

ones3()

Tensor * eve::tensor::TF::ones3	(	int	d0,
		int	d1,
		int	d2
	)

Ones 3.

在文件 TF.cpp 第 137 行定义.

引用了 d.

ones4()

Tensor * eve::tensor::TF::ones4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

Ones 4.

在文件 TF.cpp 第 141 行定义.

引用了 d.

ones5()

Tensor * eve::tensor::TF::ones5	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

Ones 5.

在文件 TF.cpp 第 145 行定义.

引用了 d.

ones6()

Tensor * eve::tensor::TF::ones6	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

Ones 6.

在文件 TF.cpp 第 149 行定义.

引用了 d.

permute2()

Tensor * eve::tensor::TF::permute2	(	Tensor *	a,
		int	a0,
		int	a1
	)

Permute 2.

在文件 TF.cpp 第 360 行定义.

引用了 a, f, order , 以及 tracing().

permute3()

Tensor * eve::tensor::TF::permute3	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2
	)

Permute 3.

在文件 TF.cpp 第 366 行定义.

引用了 a, f, order , 以及 tracing().

permute4()

Tensor * eve::tensor::TF::permute4	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3
	)

Permute 4.

在文件 TF.cpp 第 372 行定义.

引用了 a, f, order , 以及 tracing().

permute5()

Tensor * eve::tensor::TF::permute5	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3,
		int	a4
	)

Permute 5.

在文件 TF.cpp 第 378 行定义.

引用了 a, f, order , 以及 tracing().

permute6()

Tensor * eve::tensor::TF::permute6	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3,
		int	a4,
		int	a5
	)

Permute 6.

在文件 TF.cpp 第 384 行定义.

引用了 a, f, order , 以及 tracing().

popTrace()

void eve::tensor::TF::popTrace

(

Func *

f

)

Pops trace.

在文件 TF.cpp 第 77 行定义.

引用了 f.

被这些函数引用 eve::tensor::Func::compile() , 以及 eve::tensor::Func::~Func().

powScalar()

Tensor * eve::tensor::TF::powScalar	(	Tensor *	a,
		float	exp
	)

Pow scalar.

在文件 TF.cpp 第 327 行定义.

引用了 a, exp(), f, eve::tensor::PowScalar , 以及 tracing().

pushTrace()

void eve::tensor::TF::pushTrace

(

Func *

f

)

Pushes trace.

在文件 TF.cpp 第 73 行定义.

引用了 f.

被这些函数引用 eve::tensor::Func::Func().

quantizeWeight()

Tensor * eve::tensor::TF::quantizeWeight	(	Tensor *	a,
		const std::string &	dtype,
		int	group = 0
	)

Quantize weight.

Weight-only quantization (eager): pack a float32 tensor into one of the packed weight dtypes — "fp16", "fp8", "fp4", "int8", "int4". int8/int4 use symmetric per-group scales (group = elements per scale, default all).

在文件 TF.cpp 第 616 行定义.

引用了 a, eve::tensor::Float32, group, eve::tensor::q::isQuantDType(), p, eve::tensor::parseDType(), eve::tensor::q::quantize() , 以及 tracing().

rand1()

Tensor * eve::tensor::TF::rand1 ( int  d0 )

inline

Rand 1.

在文件 TF.h 第 102 行定义.

rand2()

Tensor * eve::tensor::TF::rand2 ( int  d0, int  d1  )

inline

Rand 2.

在文件 TF.h 第 104 行定义.

rand3()

Tensor * eve::tensor::TF::rand3 ( int  d0, int  d1, int  d2  )

inline

Rand 3.

在文件 TF.h 第 106 行定义.

rand4()

Tensor * eve::tensor::TF::rand4 ( int  d0, int  d1, int  d2, int  d3  )

inline

Rand 4.

Random uniform 4.

在文件 TF.h 第 108 行定义.

randn1()

Tensor * eve::tensor::TF::randn1 ( int  d0 )

inline

Randn 1.

在文件 TF.h 第 113 行定义.

randn2()

Tensor * eve::tensor::TF::randn2 ( int  d0, int  d1  )

inline

Randn 2.

在文件 TF.h 第 115 行定义.

randn3()

Tensor * eve::tensor::TF::randn3 ( int  d0, int  d1, int  d2  )

inline

Randn 3.

在文件 TF.h 第 117 行定义.

randn4()

Tensor * eve::tensor::TF::randn4 ( int  d0, int  d1, int  d2, int  d3  )

inline

Randn 4.

Random normal 4.

在文件 TF.h 第 119 行定义.

randomNormal1()

Tensor * eve::tensor::TF::randomNormal1

(

int

d0

)

Random normal 1.

在文件 TF.cpp 第 240 行定义.

引用了 t, tracing() , 以及 zeros1().

randomNormal2()

Tensor * eve::tensor::TF::randomNormal2	(	int	d0,
		int	d1
	)

Random normal 2.

在文件 TF.cpp 第 247 行定义.

引用了 t, tracing() , 以及 zeros2().

randomNormal3()

Tensor * eve::tensor::TF::randomNormal3	(	int	d0,
		int	d1,
		int	d2
	)

Random normal 3.

在文件 TF.cpp 第 254 行定义.

引用了 t, tracing() , 以及 zeros3().

randomNormal4()

Tensor * eve::tensor::TF::randomNormal4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

Random normal 4.

在文件 TF.cpp 第 261 行定义.

引用了 t, tracing() , 以及 zeros4().

randomUniform1()

Tensor * eve::tensor::TF::randomUniform1

(

int

d0

)

Random uniform 1.

在文件 TF.cpp 第 212 行定义.

引用了 t, tracing() , 以及 zeros1().

randomUniform2()

Tensor * eve::tensor::TF::randomUniform2	(	int	d0,
		int	d1
	)

Random uniform 2.

在文件 TF.cpp 第 219 行定义.

引用了 t, tracing() , 以及 zeros2().

randomUniform3()

Tensor * eve::tensor::TF::randomUniform3	(	int	d0,
		int	d1,
		int	d2
	)

Random uniform 3.

在文件 TF.cpp 第 226 行定义.

引用了 t, tracing() , 以及 zeros3().

randomUniform4()

Tensor * eve::tensor::TF::randomUniform4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

Random uniform 4.

在文件 TF.cpp 第 233 行定义.

引用了 t, tracing() , 以及 zeros4().

reduceMax()

float eve::tensor::TF::reduceMax

(

Tensor *

a

)

Reduce max.

在文件 TF.cpp 第 820 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

reduceMean()

float eve::tensor::TF::reduceMean

(

Tensor *

a

)

Reduce mean.

在文件 TF.cpp 第 804 行定义.

引用了 a, reduceSum() , 以及 tracing().

reduceMin()

float eve::tensor::TF::reduceMin

(

Tensor *

a

)

Reduce min.

在文件 TF.cpp 第 811 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

reduceSum()

float eve::tensor::TF::reduceSum

(

Tensor *

a

)

Reduce sum.

在文件 TF.cpp 第 795 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

被这些函数引用 reduceMean().

relu()

Tensor * eve::tensor::TF::relu

(

Tensor *

a

)

Relu.

reshape1()

Tensor * eve::tensor::TF::reshape1	(	Tensor *	a,
		int	d0
	)

Reshape 1.

在文件 TF.cpp 第 391 行定义.

引用了 a, d, f , 以及 tracing().

reshape2()

Tensor * eve::tensor::TF::reshape2	(	Tensor *	a,
		int	d0,
		int	d1
	)

Reshape 2.

在文件 TF.cpp 第 397 行定义.

引用了 a, d, f , 以及 tracing().

reshape3()

Tensor * eve::tensor::TF::reshape3	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2
	)

Reshape 3.

在文件 TF.cpp 第 403 行定义.

引用了 a, d, f , 以及 tracing().

reshape4()

Tensor * eve::tensor::TF::reshape4	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3
	)

Reshape 4.

在文件 TF.cpp 第 409 行定义.

引用了 a, d, f , 以及 tracing().

reshape5()

Tensor * eve::tensor::TF::reshape5	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

Reshape 5.

在文件 TF.cpp 第 415 行定义.

引用了 a, d, f , 以及 tracing().

reshape6()

Tensor * eve::tensor::TF::reshape6	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

Reshape 6.

在文件 TF.cpp 第 421 行定义.

引用了 a, d, f , 以及 tracing().

resize2d()

Tensor * eve::tensor::TF::resize2d	(	Tensor *	a,
		int	outW,
		int	outH,
		int	mode
	)

Resize 2 d.

在文件 TF.cpp 第 784 行定义.

引用了 a, f, eve::tensor::kernels::resize2d() , 以及 tracing().

rmsnorm()

Tensor * eve::tensor::TF::rmsnorm	(	Tensor *	a,
		float	eps
	)

Rmsnorm.

在文件 TF.cpp 第 507 行定义.

引用了 a, cols, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::rmsnorm(), rows , 以及 tracing().

rmsnormW()

Tensor * eve::tensor::TF::rmsnormW	(	Tensor *	a,
		Tensor *	scale,
		float	eps
	)

Rmsnorm w.

在文件 TF.cpp 第 520 行定义.

引用了 a, cols, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::rmsnorm(), rows, scale , 以及 tracing().

sdpa()

Tensor * eve::tensor::TF::sdpa	(	Tensor *	q,
		Tensor *	k,
		Tensor *	v,
		float	scale
	)

Sdpa.

在文件 TF.cpp 第 761 行定义.

引用了 q, scale, sdpaMasked() , 以及 v.

sdpaMasked()

Tensor * eve::tensor::TF::sdpaMasked	(	Tensor *	q,
		Tensor *	k,
		Tensor *	v,
		Tensor *	mask,
		float	scale
	)

Sdpa masked.

在文件 TF.cpp 第 765 行定义.

引用了 eve::tensor::Tensor::data(), f, eve::tensor::Tensor::getDim(), eve::tensor::Tensor::getRank(), mask, q, scale, eve::tensor::kernels::sdpa(), tracing() , 以及 v.

被这些函数引用 sdpa().

setRandomSeed()

void eve::tensor::TF::setRandomSeed

(

uint32_t

seed

)

Sets the random seed.

在文件 TF.cpp 第 66 行定义.

引用了 seed.

sigmoid()

Tensor * eve::tensor::TF::sigmoid

(

Tensor *

a

)

Sigmoid.

silu()

Tensor * eve::tensor::TF::silu

(

Tensor *

a

)

Silu.

sin()

Tensor * eve::tensor::TF::sin

(

Tensor *

a

)

Sin.

slice()

Tensor * eve::tensor::TF::slice	(	Tensor *	a,
		int	axis,
		int	begin,
		int	end
	)

Slice.

在文件 TF.cpp 第 689 行定义.

引用了 a, begin, end, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::sliceOp() , 以及 tracing().

softmax()

Tensor * eve::tensor::TF::softmax	(	Tensor *	a,
		int	axis
	)

Softmax.

在文件 TF.cpp 第 455 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::softmax() , 以及 tracing().

sqrt()

Tensor * eve::tensor::TF::sqrt

(

Tensor *

a

)

Sqrt.

sub()

Tensor * eve::tensor::TF::sub	(	Tensor *	a,
		Tensor *	b
	)

Sub.

subScalar()

Tensor * eve::tensor::TF::subScalar	(	Tensor *	a,
		float	s
	)

Sub scalar.

在文件 TF.cpp 第 312 行定义.

引用了 a, f, s, eve::tensor::SubScalar , 以及 tracing().

sumAxis()

Tensor * eve::tensor::TF::sumAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

Sum axis.

tanh()

Tensor * eve::tensor::TF::tanh

(

Tensor *

a

)

Tanh.

tracing()

Func * eve::tensor::TF::tracing

(

)

const

Tracing.

在文件 TF.cpp 第 91 行定义.

被这些函数引用 addScalar(), arange(), argmax(), avgpool2d(), cast(), clamp(), concatN(), conv1dBias(), conv2dBias(), divScalar(), embedding(), eye(), flatten(), layernorm(), layernormWB(), linspace(), logSoftmax(), matmul(), maximumScalar(), maxpool2d(), minimumScalar(), mulScalar(), permute2(), permute3(), permute4(), permute5(), permute6(), powScalar(), quantizeWeight(), randomNormal1(), randomNormal2(), randomNormal3(), randomNormal4(), randomUniform1(), randomUniform2(), randomUniform3(), randomUniform4(), reduceMax(), reduceMean(), reduceMin(), reduceSum(), reshape1(), reshape2(), reshape3(), reshape4(), reshape5(), reshape6(), resize2d(), rmsnorm(), rmsnormW(), sdpaMasked(), slice(), softmax(), subScalar(), transpose() , 以及 where().

transpose()

Tensor * eve::tensor::TF::transpose

(

Tensor *

a

)

Transpose.

在文件 TF.cpp 第 354 行定义.

引用了 a, f , 以及 tracing().

where()

Tensor * eve::tensor::TF::where	(	Tensor *	cond,
		Tensor *	a,
		Tensor *	b
	)

Where.

在文件 TF.cpp 第 435 行定义.

引用了 a, b, f, eve::tensor::Tensor::get(), eve::tensor::Tensor::getDim(), eve::tensor::Tensor::getRank(), eve::tensor::Tensor::getSize(), eve::tensor::Tensor::kMaxRank, tracing() , 以及 eve::tensor::Where.

zeros1()

Tensor * eve::tensor::TF::zeros1

(

int

d0

)

Zeros 1.

在文件 TF.cpp 第 104 行定义.

引用了 d.

被这些函数引用 randomNormal1() , 以及 randomUniform1().

zeros2()

Tensor * eve::tensor::TF::zeros2	(	int	d0,
		int	d1
	)

Zeros 2.

在文件 TF.cpp 第 108 行定义.

引用了 d.

被这些函数引用 randomNormal2() , 以及 randomUniform2().

zeros3()

Tensor * eve::tensor::TF::zeros3	(	int	d0,
		int	d1,
		int	d2
	)

Zeros 3.

在文件 TF.cpp 第 112 行定义.

引用了 d.

被这些函数引用 randomNormal3() , 以及 randomUniform3().

zeros4()

Tensor * eve::tensor::TF::zeros4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

Zeros 4.

在文件 TF.cpp 第 116 行定义.

引用了 d.

被这些函数引用 randomNormal4() , 以及 randomUniform4().

zeros5()

Tensor * eve::tensor::TF::zeros5	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

Zeros 5.

在文件 TF.cpp 第 120 行定义.

引用了 d.

zeros6()

Tensor * eve::tensor::TF::zeros6	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

Zeros 6.

在文件 TF.cpp 第 124 行定义.

引用了 d.

---

该类的文档由以下文件生成:

• src/modules/tensor/TF.h
• src/modules/tensor/TF.cpp