Tensorflow vs Pytorch 명령어 비교 -(6)
tf.zeros(shape, dtype=tf.dtypes.float32, name=None) with tf.Session() as sess: print(sess.run(tf.zeros([3, 4], tf.int32))) print(sess.run(tf.zeros([3, 4], tf.float32))) [[0 0 0 0] [0 0 0 0] [0 0 0 0]] [[0. 0. 0. 0.] [0. 0. 0. 0.] [0. 0. 0. 0.]] torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor torch.zeros([3, 4], dtype=torch.int32) torch..
2021. 6. 3.
Tensorflow vs Pytorch 명령어 비교 -(5)
tf.gather(params, indices, validate_indices=None, name=None, axis=None, batch_dims=0) v1 = tf.constant([1, 3, 5, 7, 9, 0, 2, 4, 6, 8]) v2 = tf.constant([[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]]) with tf.Session() as sess: print(sess.run(tf.gather(v1, [2, 5, 2, 5], axis=0))) print(sess.run(tf.gather(v2, [0, 1], axis=0))) print(sess.run(tf.gather(v2, [0, 1], axis=1))) [5 0 5 0] [[ 1 2 3 4 5 6] [..
2021. 6. 2.