def __array(array, dtype, is_volatile, transposition):
volatile = "ON" if is_volatile else "OFF"
if not transposition:
return Variable(XP.__lib.array(array, dtype=dtype), volatile=volatile)
else:
return F.swapaxes(Variable(XP.__lib.array(array, dtype=dtype), volatile=volatile), 0, 1)
评论列表
文章目录