The equal function for THCudaBlas_Sgemm in Aten

BestSonny · August 21, 2018, 7:10am

THCudaBlas_Sgemm(state, ‘n’, ‘n’, n, m, k, 1.0f,
THCudaTensor_data(state, columns), n,
THCudaTensor_data(state, weight), k, 1.0f,
THCudaTensor_data(state, output_n), n);