开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

实现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]
[object Object]
[object Object]
  • self[object Object]):4维张量,shape为(N, C, H, W)。
  • x1[object Object]):4维张量,shape为(N, C, H, W)。
[object Object]
  • [object Object][object Object]的shape相同。
  • H和W是[128, 256, 384, 512]其中之一。
  • (N * C)%32结果为0。
  • [object Object]为-1。
[object Object]
[object Object]