实现axpy_v2、softmax_v2、drop_out_domask_v3功能。即:
y=x1+ self *alpha
Softmax(xi)= exp(xi)/∑jexp(xj)
output = 根据mask舍弃x中的元素,留下来的元素乘(1/prob)
[object Object]
- self(
[object Object]):4维张量,shape为(N, C, H, W)。 - x1(
[object Object]):4维张量,shape为(N, C, H, W)。
[object Object]和[object Object]的shape相同。- H和W是[128, 256, 384, 512]其中之一。
- (N * C)%32结果为0。
[object Object]为-1。
[object Object]