|
|
@@ -601,7 +601,15 @@ namespace Tensorflow |
|
|
|
}
|
|
|
|
|
|
|
|
public static Tensor gather<T1, T2>(T1 @params, T2 indices, string name = null, int axis = 0)
|
|
|
|
=> gen_array_ops.gather_v2(@params, indices, axis, name: name);
|
|
|
|
{
|
|
|
|
if (axis != 0)
|
|
|
|
return gen_array_ops.gather_v2(@params, indices, axis, name: name);
|
|
|
|
|
|
|
|
if (@params is ResourceVariable variable)
|
|
|
|
return variable.sparse_read();
|
|
|
|
|
|
|
|
return gen_array_ops.gather_v2(@params, indices, axis, name: name);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static Tensor transpose<T1, T2>(T1 a, T2 perm, string name = "transpose", bool conjugate = false)
|
|
|
|
{
|
|
|
|