corn_bag Help

tensor

张量, 类似np 的ndarray, tensor 能利用gpu 以及能做反向误差求导.
这两个特性对机器学习很重要.

data = [[1,2,3],[4,5,6]] arr = array (data) #新建张量 tns = tensor(data) #tensor([[1, 2, 3], # [4, 5, 6]]) # 取一行 tns[1] # 取一列 tns[:,1] # 取一块 tns[r_s:r_e,c_s:c_e] #[b,e) 前闭后开的约定. # 数加, 数乘 tns + 1 tns * 1.5 #类似c ,自动转换到float 张量. # type 返回元素类型. tns.type()
Last modified: 22 七月 2024