MindSpore中张量的属性和运算

张量之间有很多运算,包括算术、线性代数、矩阵处理(转置、标引、切片)、采样等,下面介绍其中几种操作,张量运算和NumPy的使用方式类似。

  1. 标准的类似NumPy的索引和切片:

    tensor = Tensor(np.array([[0, 1], [2, 3]]).astype(np.float32))

    print("First row: {}".format(tensor[0]))

    print("First column: {}".format(tensor[:, 0]))

    print("Last column: {}".format(tensor[..., -1]))

  2. Concat将给定维度上的一系列张量连接起来。

    data1 = Tensor(np.array([[0, 1], [2, 3]]).astype(np.float32))

    data2 = Tensor(np.array([[4, 5], [6, 7]]).astype(np.float32))

    op = ops.Concat()

    output = op((data1, data2))

    print(output)

  3. Stack则是从另一个维度上将两个张量合并起来。

    data1 = Tensor(np.array([[0, 1], [2, 3]]).astype(np.float32))

    data2 = Tensor(np.array([[4, 5], [6, 7]]).astype(np.float32))

    op = ops.Stack()

    output = op([data1, data2])

    print(output)

  4. 普通运算:

    input_x = Tensor(np.array([1.0, 2.0, 3.0]), mstype.float32)

    input_y = Tensor(np.array([4.0, 5.0, 6.0]), mstype.float32)

    mul = ops.Mul()

    output = mul(input_x, input_y)

    print(output)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值