- 博客(2)
- 收藏
- 关注
RSS订阅原创 torch.squeeze / torch.unsqueeze
torch.squeeze / torch.unsqueeze 消除tensor维度数为 1 的维度，添加维度数为 1 的维度。 1- 首先squeeze，当且仅当tensor存在维度数为 1 时，生效。先从二维来考虑。 当tensor的形状为1*3时， a = torch.randn(1,3) print ( a ) b = torch.squeeze( a , dim = 0) # dim =0对应（1,3）中的 1 print ( b ) 显示结果： a = [ [ 1 ...
2021-05-10 11:07:07
148
原创 CALLBACK / 回调（diao）
自己写代码很少用到这个功能（自己太低端），但是学习深度学习时，经常阅读别人的代码，大佬经常使用此功能。 没法，只得去学习学习啦。在网络上搜索了很多帖子，学习了大佬的经验。下面记录一下学习心得。 callback发生在主程序（main function）调用库函数（library function），而库函数又去调用我们自定义的函数（user function）的过程中。 那这跟callback有什么关系呢？ 原来 大佬们通常把main function所在的层称之为高层，library funct
2021-05-08 12:36:41
247
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人