摘要: 1.一次并行计算一个batch,因此所有的计算过程产生的tensor多了第一维batch_size 2.源码实现大量用到了python的广播功能(broadcasting),实现了不同维度数组间的直接计算 3.tensor之间的各种拼接变换组合 4.还有经常用的tensor.max()函数,可以输出 阅读全文
posted @ 2020-09-02 00:11 Parallax 阅读(169) 评论(0) 推荐(0) 编辑