稀疏张量#

Mars张量支持稀疏张量,不幸的是,目前只能使用2维稀疏张量。多维稀疏张量将在后续支持。

创建稀疏张量的函数#

mars.tensor.tensor

mars.tensor.zeros

返回一个具有给定形状和类型的新张量,填充为零。

mars.tensor.eye

返回一个二维张量,对角线上的元素为1,其余元素为0。

mars.tensor.identity

返回单位张量。

mars.tensor.random.randint

返回从 low(包含)到 high(不包含)的随机整数。