参考https://github.com/chenyuntc/pytorch-book/tree/v1.0 希望大家直接到上面的网址去查看代码,下面是本人的笔记 Tensor Tensor可以是一个数(标量).一维数组(向量).二维数组(矩阵)或更高维的数组(高阶数据) Tensor和numpy的ndarrays类似,不同在于pytorch的tensor支持GPU加速 导包: from __future__ import print_function import torch as t 判断是否
一.基础函数 1.1 .tf.reduce_sum(input_tensor, axis) Computes the sum of elements across dimensions of a tensor,沿着维度sxis计算和 x= [[, , ], [, , ]],其秩为2 //求和,在所有维度操作,也就相当于对所有元素求和 tf.reduce_sum(x) ==> //在维度0上操作,在这个例子中实际就是按列(维度0)求和 tf.reduce_sum(x, ) ==> [, ,